tensorflow 怎样确定是cpu 还是 gpu?

Question

tensorflow 怎样确定是cpu 还是 gpu

Answer 1

安装GPU版本的tensorflow就可以了pip install --upgrade 还需要安装GPU加速包，下载 CUDA 和 cuDNN 的安装包了，注意版本号分别是 CUDA 8.0 和 cuDNN 5.1。如果都安装成功，运行程序时会自动使用GPU计算