Win10安装tensorflow-gpu

Win10安装tensorflow-gpu

• 选择正确的版本
• CUDA 10.0
• cuDNN
• 更改环境变量
• tensorflow-gpu
• 验证

win10系统+gtx1050，试水装一下tensorflow-gpu版，为下次回学校装台式机的环境做准备，记录贴。

选择正确的版本

1. 选择正确的版本很重要。之前装了CUDA 9.0，但是tensorflow-gpu是1.15，版本不匹配导致一直报错。
2. 首先看一下GPU驱动的版本，然后根据驱动版本选择CUDA版本。驱动版本太低可以更新一下。之前为了配合驱动版本装了CUDA 9.0，结果tensorflow-gpu要不速度太慢下不下来，速度快的只找到1.14以上的版本，所以最后去升级驱动，改装CUDA 10.0。下图是更新后的驱动版本为441.45。
3. 根据对应的表格，最后选择CUDA 10.0 + cudnn 7.6.5 + tensorflow-gpu 1.15。

CUDA 10.0

1. 到网站下载: CUDA 10.0.
2. 选择默认路径+精简模式安装即可。（便于后续改环境路径）
   

cuDNN

1. 到网站下载: cuDNN.
2. 这里选择v7.6.5对应CUDA 10.0的。
3. 将cuDNN里面对应文件夹bin, include, lib里的内容贴到CUDA\V10.0对应的文件夹里。
   

更改环境变量

1. 双击环境变量的path, 添加下列内容。
   

tensorflow-gpu

1. 直接安装非常慢，选择了清华的源就很快：
   pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.15
2. 最后的1.15是版本，不写版本似乎默认是下载最新的

验证

参考链接：验证.