一、电脑是否支持GPU版本
查看方法:
windows + R
在cmd中输入dxdiag
可以看到我的显2是NVIDIA生产的GeForce系列的
这个时候就可以去NVIDIA官网去下载自己芯片对应的版本
我的是GTX 1050 所以找相对应的
NVIDIA官网
二、安装TensorFlow-gpu
首先我的Python版本是Python 3.7的,TensorFlow是可以支持的。
安装TensorFlow-gpu命令
pip install tensorflow-gpu -i 加速源
三、安装CUDA和CUDDN
TensorFlow和cuda以及cuDDN也有版本对应关系的,所以在安装的时候先查看自己的TensorFlow的版本。
在命令行中输入:pip list
可以看到我的TensorFlow版本是2.0.0