tensorflow-gpu的安装

tensorflow-gpu安装

Win10，anaconda3/python.exe（3.7版），使用的是anaconda里的python，不需要再自己安装python。如果有GPU可以安装tensorflow-gpu，没有GPU就安装tensorflow
下是讲tensorflow-gpu的安装（2020-3-9）：
最重要的是CUDA，cuDNN，tensorflow-gpu版本一定匹配，这个必须要搞清楚不然后面还得卸载。https://www.tensorflow.org/install/source_windows鼠标滑到底，有对应的版本。
我的版本是CUDA（10.0）+cuDNN（7.6.5）+tensorflow-gpu（2.0）
CUDA的安装：https://developer.nvidia.com/cuda-toolkit-archive：这个是英伟达CUDA的历史版本，找到需要的版本号，不要直接下载最新版的安装包。下载安装包后一切默认安装即可。安装完成后可通过cmd里面输入nvcc -V查看CUDA版本号。
cuDNN的安装：英伟达官网下载: https://developer.nvidia.com/rdp/cudnn-download 下载对应的版本，需要注册，可以用微信号或者qq注册。
下载完成解压缩之后，把bin，include，lib/x64里的文件分别放入CUDA对应的bin，include，lib/x64里面。CUDA的默认路径为：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
利用清华源安装tensorflow-gpu:

pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

完美！