按照这个教程装的,记录在这里:
https://blog.csdn.net/weixin_36368407/article/details/54177380
这个帮助内容也很好:
http://www.netinstructions.com/how-to-install-and-run-gpu-enabled-tensorflow-on-windows/
略有强迫症,看着低版本不爽,必须装最新的。。
所以按照教程步骤做,但是都用尽量新的版本。
安装CUDA完,记得要配置环境变量。
默认路径:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
访问bin和lib\x64下的程序包
path中加上这两个路径即可:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
- nvidia geforce 940mx
- 核显 Intel(R) HD Graphics 520
- windows 10
- python 3.6.4
- tensorflow-gpu-1.7.0
- CUDA 9.0
注意:CUDA 8.0失败后,又安装的CUDA 9.1,不匹配,已踩坑。。。重新下载的CUDA 9.0。。。
Could not find 'cudart64_90.dll'
CUDA.jpg
测试效果:
TIM截图20180412040912.jpg
网友评论