1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > win10 GTX1050配置VS19+CUDA10.2+CUDNN

win10 GTX1050配置VS19+CUDA10.2+CUDNN

时间:2020-09-09 09:13:11

相关推荐

win10 GTX1050配置VS19+CUDA10.2+CUDNN

需要注意的前提条件

我做了这几个先验条件,全程安装没有踩到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

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。