Windows安装GPU的tensorflow

之前一直装不下,后改道去了Linux,结果ubuntu的N卡驱动居然报错了,手贱点了更新可能,就导致不能用GPU ,现在想不每次更换系统,直接在windows下装个GPU版的吧。

Windows安装CPU很方面,直接

pip install tensorflow

会根据你自己本身的python来安装比较新的tensorflow。

原本装了CPU版的,现在改成GPU的,先卸载原来的tensorflow。

pip uninstall tensorflow

接下来就是重点了!!

1. 确立自己要装的哪个CUDA,GPU兼容问题,支持的windows版本,还有确立支持的VS

  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可能不匹配。  参考

2. 装 VS

确立好要装的东西至关重要。

安装的网址: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,没有这么容量怎么办,参考我另外一个文,地址

3. 装CUDA

选择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,一样的操作。

4. 装Cudnn

Cuda9.2对应下载7.2.1

刚开始选择了低一些的版本7.1.4,然后成功给我报错说版本太低,哦吼~

 解压下载的文件,将文件夹里的内容拷贝到CUDA的安装目录并覆盖相应的文件夹

CUDA拷贝目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

5. 安装tensorflow

我选择的1.10.0,其他的也是可以的,

https://github.com/fo40225/tensorflow-windows-wheel  这里可以选择对应的版本

千万不要直接pip install tensorflow-gpu这样,要用上面的whl文件安装。

最后,结束!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值