一开始我只装了tensorflow,没有装tensorflow-gpu,先pip install 安装一下tensorflow-gpu
pip install tensorflow-gpu==1.15.0 -i https://pypi.doubanio.com/simple/
血与泪的教训告诉我不要装高版本的比如tensorflow2以及低版本的比如1.10
然后python命令行输入
import tensorflow as tf
tf.__version__
如果显示的是tensorflow-gpu的版本而不是tensorflow就行(tensorflow-gpu的版本需要高于tensorflow,python默认使用版本更高的tensorflow,所以可以用anaconda创建新环境更方便一点)
1.安装CUDA
CUDA下载地址:https://developer.nvidia.com/cuda-downloads
CUDA历史版本:https://developer.nvidia.com/cuda-toolkit-archive
安装直接选择精简模式就行(全家桶模式),解压路径随意,安装路径建议默认就好
然后下载cuDNN:https://developer.nvidia.co