win10系统+gtx1050,试水装一下tensorflow-gpu版,为下次回学校装台式机的环境做准备,记录贴。
选择正确的版本
- 选择正确的版本很重要。之前装了CUDA 9.0,但是tensorflow-gpu是1.15,版本不匹配导致一直报错。
- 首先看一下GPU驱动的版本,然后根据驱动版本选择CUDA版本。驱动版本太低可以更新一下。之前为了配合驱动版本装了CUDA 9.0,结果tensorflow-gpu要不速度太慢下不下来,速度快的只找到1.14以上的版本,所以最后去升级驱动,改装CUDA 10.0。下图是更新后的驱动版本为441.45。
- 根据对应的表格,最后选择CUDA 10.0 + cudnn 7.6.5 + tensorflow-gpu 1.15。
CUDA 10.0
- 到网站下载: CUDA 10.0.
- 选择默认路径+精简模式安装即可。(便于后续改环境路径)
cuDNN
- 到网站下载: cuDNN.
- 这里选择v7.6.5对应CUDA 10.0的。
- 将cuDNN里面对应文件夹bin, include, lib里的内容贴到CUDA\V10.0对应的文件夹里。
更改环境变量
- 双击环境变量的path, 添加下列内容。
tensorflow-gpu
- 直接安装非常慢,选择了清华的源就很快:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.15 - 最后的1.15是版本,不写版本似乎默认是下载最新的
验证
参考链接:验证.