Win 10 安装NVIDIA cuda cudnn tensorflow-gpu 2.0
安装NVIDIA
最近想入门一下深度学习,所以就先更新了一下显卡
- NVIDIA类型 :GeForce 940MX;
- 更新后的驱动程序版本:456.81;
NVIDIA download:link
下载过程很简单,直接下一步即可,下载时间取决于网络通畅程度
踩坑:更新之后,桌面右击的菜单栏中没有NVIDIA控制面板了,下载了NVIDIA控制面板后一直显示没有连接GPU显示器…
解决方法:看了许多网上的解决办法,但在我这没啥用emmm,在这统一汇总一下:
方法一
我的电脑–>管理–>设备管理器–>显示适配器,查看适配器下方的选项是否启动,如果未启动,将其改为启动 或者 直接重新启动
方法二
电脑管家–>查看是否将NVIDIA列入禁用启动项,若列入禁用启动项,则将其启动
方法三
win+R–>mscofig->选择引导标签页->打开高级选项设置->取消CPI锁定
方法四
尝试了上述方法无果,之后在我的电脑(右击)->管理->设备管理器->NIVIDIA GEFORCE 940MX属性中点击了回退驱动程序,回退结束之后又更新了一次,这时发现右击菜单栏中已经有了NVIDIA控制面板,并且不再显示未连接GPU显示器
搞了大半个一晚上,心情奔溃…
安装cuda
打开NVIDIA控制面板,帮助选项卡->点击系统信息,可以看到当前的驱动程序版本为456.81,找到和其相对应的cuda 版本,可以看到我应该装一个cuda11.1
NVIDIA与cuda版本的对应表如下:link
cuda download:link
下载可能会比较慢,我看网上有朋友在下载时选择了自定义下载,跳过了visial stutio的环境安装,这样可能会快一点。但我当时网速还可以,就没有选择自定义下载,大约十分钟左右下载完成。
win+R->nvidia-smi 查看是否安装成功
因为后面要装tensorflow-gpu,发现tensorflow-gpu好像只支持到cuda10,所以又下载了cuda10…
将cuda10设置为默认
安装cuda完成后,安装程序会自动帮助你修改系统环境变量。如果要将cuda10 设置为默认使用的cuda,就将cuda_path设置为10.0,并且下方10.0的cuda_path要在11.1上面
cmd中输入nvcc --version,查看版本,发现默认cuda修改为cuda10.0
安装cudnn
cuDNN Download:link
- 我安装的是cudnn 7.5.0。压缩包下载完成之后,将压缩包中的bin include lib文件夹复制到cuda对应的位置(位置见下图)
配置环境变量
- CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0
- CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
- CUDA_BIN_PATH = %CUDA_PATH%\bin
- CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
- CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x6
在环境变量path选项中添加:
1.%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;
2.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64;
3.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin;
4.C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0\common\lib\x64;
5.C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0\bin\win64;
win+R,输入 cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite ,然后分别执行bandwidthTest.exe和deviceQuery.exe,得到两个PASS,就说明安装成功啦!
补:8.0~11.0 cuda +cudnn 集合
链接:https://pan.baidu.com/s/1hLY0XDh4F6_GLIL_r30XIg
提取码:mce9
安装 tensorflow-gpu 2.0
- 如果之前有install过tensorflow,建议先卸载之前的再下载tensorflow-gpu,tensorflow默认使用的是cpu,非gpu
- win+R->cmd->pip install tensorflow-gpu==2.0
因为之前修改了pip.ini 文件,所以默认使用了豆瓣的镜像
安装完成后,启动jupyter
import tensorflow as tf
print(tf.test.is_gpu_avaliable()
显示True,则说明安装成功
补:默认使用镜像的方法(无需每次pip install 都输入镜像源名字)
- C:\Users\用户名\下新建pip文件夹
- 在pip 文件夹中新建pip.ini文件
- 在pip.ini 文件中输入:
[global]
index-url = https://pypi.douban.com/simple
即可