Win10安装tensorflow-gpu=1.14+CUDA=10.0+cudnn=10.0

因为用的是tensorflow的gpu版本,要用到NVIDIA的显卡,需要再安装与tensorflow和显卡相匹配的CUDA和cuDNN,一定注意要三者相匹配的版本,如果错装了其他版本的CUDA和cuDNN,建议卸载了,重装正确的版本。

下面说下,如何确定应该安装的CUDA和cuDNN版本,假设tensorflow安装在F:\Technicalware\Anaconda3\envs\tensorflow目录下,
那么以文本方式打开F:\Technicalware\Anaconda3\envs\tensorflow\Lib\site-packages\tensorflow\python\platform\build_info.py这个文件,可以看到你的tensorflow所应该使用的CUDA 和cuDNN 版本:

Win10安装tensorflow-gpu=1.14+CUDA=10.0+cudnn=10.0_第1张图片

这里的msvcp140.dll应该在C:\Windows\System32\下面,是通过安装vc 2005 redistribute提供的,nvcuda.dll也是在C:\Windows\System32\下面,cudart64_100.dll则是安装CUDA 10.0时产生的,cudnn64_7.dll是安装cuDNN 7时产生的,注意cudnn的版本要与CUDA版本对应,任何一个文件缺失或者版本不对都可能会导致tensorflow在import时报ImportError错。

接下来就正式开始安装

第一步,安装tensorflow1.14

pip install tensorflow-gpu==1.14

第二步,查看对应版本

第三步,安装CUDA
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork

第四步,安装cudnn
https://developer.nvidia.com/rdp/cudnn-download
把cudnn-10.0-windows10-x64-v7.6.1.34.zip解压后,把bin,include,lib三个文件夹直接复制到目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\下即可。

第五步,添加相应的环境变量
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin

第六步,测试import tensorflow

hello=tensorflow.constant('hello')

总结一下遇到的问题

问题一:tensorflow-gpu=1.14,CUDA,cudnn版本不匹配导致的问题
解决方案:安装对应版本

问题二:卸载CUDA后重新安装出现问题
解决方案:
1、删除系统盘中Program Files (x86)和Program Files内包含的NVIDIA Corporation和NVIDIA GPU Computing Toolkit文件夹,删除注册表有关内容(一定要清理干净!!!)
2、重启电脑,然后重新安装cuda,击自定义,转到另一个可以自定义功能的界面,把vs相关的那一项的勾选去掉,然后就正常安装了,最后显示安装成功!

参考文献
https://blog.csdn.net/qq_35077107/article/details/94555320
https://blog.csdn.net/w670165403/article/details/81737139
https://blog.csdn.net/zzpong/article/details/80282814?depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-2&utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-2

你可能感兴趣的:(python基础)