需要注意的前提条件
我做了这几个先验条件,全程安装没有踩到bug,很顺利~
1. 需要卸载显卡驱动:link
2. 有以前的CUDA版本的话需要卸载,在控制面板-卸载程序中,
清除带有CUDA的所有东西、
再清除除了NVIDIA 图形驱动程序、NVIDIA PhysX 系统软件、NVIDIA GeForce Experience(只保留这三个)之外的NVIDIA的东西
3.需要删除以前的VS版本,这里是VS来装的,卸载我没有做,可以参考link
4.此外,还需要查询GPU版本是否能够被CUDA所支持。
1.VS安装
进入官网:VS
点击Visual Studio,我下载的Community版本,一路点击,安装后打开installer。
注意:只选择 使用C++桌面开发 这个选项,基本就能满足要求,我是安装的默认路径
安装之后重启,进入CUDA安装部分:
2.CUDA10.2安装
进入/cuda-downloads
选择相应的CUDA版本进行安装。
在安装中,注意选择自定义,勾选所有选项,一路点击即可。(这一部分会自动安装显卡驱动)
这里安装它自动添加了环境变量,可以到高级系统变量-PATH中查得到。
测试:在cmd中输入nvcc -V查看是否有版本信息,有则安装成功
3.CUDNN
/rdp/cudnn-download
勾选I agree 下载对应的版本
解压后是三个文件夹,将其粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2文件夹中。
测试
至此,安装完毕
可以通过在python环境中(anaconda虚拟环境也OK)
import torchtemp=torch.Tensor([3.])temp.cuda()
若不报错则成功安装。
借鉴blog:
/houzupi2534/article/details/100010836
/article/2481243398/
/guoyaohua/p/9265268.html