之前一直装不下,后改道去了Linux,结果ubuntu的N卡驱动居然报错了,手贱点了更新可能,就导致不能用GPU ,现在想不每次更换系统,直接在windows下装个GPU版的吧。
Windows安装CPU很方面,直接
pip install tensorflow
会根据你自己本身的python来安装比较新的tensorflow。
原本装了CPU版的,现在改成GPU的,先卸载原来的tensorflow。
pip uninstall tensorflow
接下来就是重点了!!
https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
虽然截取的图片是10的,但是也有参考价值。
我确立的安装是VS2017,CUDA9.2,,Cudnn7这个时候不要选择最新的了。
参考别的博主有
python3.5+VS2015+CUDA8.0+CUDnn6.0
CUDA9.2 + cuDNNv7.2.1 + tensorflow-gpu 1.9.0
vs2017+CUDA9.2+tensorflow-gpu1.8.0
总结以下问题:
1. CUDA8.0不支持VS2017
2. Tensorflow1.6开始使用CUDA9.0+cuDNN7
3. CUDA9.2,版本太高跟tensorflow1.8不匹配
4. 亲测Tensorflow1.10.0跟CUDA9.2也不匹配
5. 如果使用tensorflow官方源码安装,还是选择CUDA9.0 稳妥,9.2可能不匹配。 参考
确立好要装的东西至关重要。
安装的网址:https://docs.microsoft.com/en-us/visualstudio/productinfo/installing-an-earlier-release-of-vs2017#top
百度云资源 链接:https://pan.baidu.com/s/18ypSIYU2UAWj-xj-V8iVSw 提取码:ju5f
在安装过程勾选哪些可参考上面博主的第三个。
另外要提的就是VS要装在C盘,但容量很大,要25个G,没有这么容量怎么办,参考我另外一个文,地址
选择9.2 下载
其中两个都下载下来。
一定要用管理员权限安装。
先解压上面那个,找到setup.exe安装,用自定义安装,勾选CUDA里的Development、Runtime、Documentation、Samples。绝对不要勾选Visual Studio Integration。另外的driver,other什么的想装就装,驱动本身系统是带了的。
安装完后,进入到解压后路径,选择 \cuda_9.2.88_win10\CUDAVisualStudioIntegration\extras\visual_studio_integration\MSBuildExtensions,把这路径的文件复制到VS下C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\Common7\IDE\VC\VCTargets\BuildCustomizations,这一步好像只是为了让VS跑的通sample。反正我不行,
只要在cmd里输入 nvcc-c,有
就表示安装成功了。
接着安装第二个patch,一样的操作。
Cuda9.2对应下载7.2.1
刚开始选择了低一些的版本7.1.4,然后成功给我报错说版本太低,哦吼~
解压下载的文件,将文件夹里的内容拷贝到CUDA的安装目录并覆盖相应的文件夹
CUDA拷贝目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2
我选择的1.10.0,其他的也是可以的,
https://github.com/fo40225/tensorflow-windows-wheel 这里可以选择对应的版本
千万不要直接pip install tensorflow-gpu这样,要用上面的whl文件安装。
最后,结束!