一. 安装环境
Thinkpad T590/ 联想天逸510pro,系统 Windows 10 64bit 家庭中文版
GPU: Nvidia MX250(CUDA10.0以下存在兼容性问题) / Nvidia GeForce GT 730
Python: 3.7.3(Anaconda3-2019.07-Windows-x86_64)
CUDA: cuda_10.0.130_411.31_win10.exe
cuDNN: cudnn-10.0-windows10-x64-v7.6.5.32.zip
Q1:显卡驱动版本号及其对应的CUDA版本
关于显卡驱动的查询,可参考’nvidia-smi’ 不是内部或外部命令,也不是可运行的程序。
或者 管理–>设备管理器–>显示适配器–>选中显卡,属性,即可看到驱动程序版本,比如MX250的26.21.14.3140,对应的就是431.40,Nvidia GeForce GT 730的27.21.14.5752对应的就是457.52
也可以右键–>Nvidia控制面板,
结合对比图,可以看到cuda10.1可以满足要求。
Q2:tensorflow版本号及其对应的CUDA版本
Tensorflow CUDA及CUDNN版本对应关系表查询:遇事不决请参考官方说明
这里值得一提的是,Python,CUDA,cuDNN之间的版本要严格匹配,不匹配安装会出错。
二. 安装
1. 安装visual studio
VS2015及以上,本人安装的是VS2015community,按照上图的指示。这里最好安装VS2019。
2. 安装cuda和cudnn
cuda直接点击安装,默认即可。安装好以后会有以下文件夹
然后解压cudnn,将cuda文件下
的所有内容拷贝到v10.1文件夹内,如需替换,直接替换(本人安装时无需替换)。
安装anaconda
安装时默认不勾选添加路径操作,安装完毕后将安装目录下的…\Anaconda3, …\Anaconda3\Scripts, …\Anaconda3\Library\bin三个文件夹添加到环境变量的path中
安装tensorflow_gpu
win+R打开cmd, 然后
pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ tensorflow_gpu==2.1.0
正常情况下会下载很快,然后安装成功
然后是见证bug的时刻
Bug1: ImportError: DLL load failed: 找不到指定的模块
原因:安装2.1.0版本程序包需要msvcp140_1.dll文件,可能跟VS版本较低有关,这里直接安装VC_redist.x64.exe文件即可。
ImportError: DLL load failed: 找不到指定的模块
Bug2: Could not load dynamic library ‘cudart64_101.dll’; dlerror: cudart64_101.dll not found
在目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1下搜索cudart64,可能为cudart64_100或102,直接重命名为cudart64_101.dll即可。
Could not load dynamic library ‘cudart64_101.dll’; dlerror: cudart64_101.dll not found解决方法
恭喜时刻
安装pycharm(optional)
默认即可,最多更换一下路径。