参考:https://www.cnblogs.com/guoyaohua/p/9265268.html
Install Anaconda
- download newest Anaconda from Web.(I use Anaconda3-2019.10-Windows-x86_64.exe )
Install TensorFlow GPU
- run anaconda prompt
conda create --name tf-gpu python=3.6
activate tf-gpu
conda install tensorflow-gpu
,此命令网络不佳的情况下可能失败,重复几次最终可以安装完成。
安装GPU相关驱动和软件
- 访问TensorFlow官网https://www.tensorflow.org/install/gpu/,查看支持的显卡/CUDAToolkit以及cuDNNsdk版本。
- 进入Nvidia官网下载软件,注意Toolkit安装程序可以直接下载,sdk需要注册用户后下载。
- Toolkit自定义全选安装即可,cuDNNsdk只需解压后复制到Toolkit安装目录下对应的文件夹。
- 我的笔记本显卡是GeforceGTX960M,支持CUDA,下载了最新的Toolkit10.1和sdk7.6.5
检查GPU是否工作
import tensorflow as tf
tf.test.gpu_device_name()
- 如输出GPU字样即为OK