pytorch 卸载_windows安装cuda和cudnn以及pytorch+卸载

论坛 期权论坛 编程之家     
选择匿名的用户   2021-5-30 14:32   47   0

查看cuda版本号

1、首先需要进入pytorch官网查看一下需要安装的pytorch版本适配的cuda版本号:

网址如下所示:

PyTorchpytorch.org
3ad1e7fe6364f0a10ba66250954c1dec.png

如图所示,官网默认显示最新版本的PyTorch:

a804223dc511aec8383408ddfaf8328e.png

点击下面的链接,可以安装一些老PyTorch的版本:

83f972cdc9bab926491f253bb721f8a4.png

点击上面链接后,出现如下页面:

f7e61afc21e1c58479626bb01c71f115.png

最后,根据自己的要求找到对应版本的PyTorch,找到PyTorch对应的cuda版本。下面开始进行cuda和cudnn的安装。

cuda以及cudnn安装

1、首先在桌面用鼠标右键进行右击,会显示nvidia面板,选择这个选项,

(1)如果没有这个选项,需要进行安装nvidia驱动器(直接点击下列链接进行安装即可)。如果有这个选项,则不需要进行安装,直接跳过本步骤即可。

知乎 - 安全中心www.nvidia.cn

(2)点击上面链接之后,显示如下页面,会自动识别显卡型号,如下图所示

bbf683ac31db4344a5edf2b667122988.png

(3)然后点击搜索再点击下载就行,然后就一步步安装就行

2、在电脑有了nvidia驱动器以后,查看nvidia驱动器支持的cuda版本:

3、在电脑桌面右键,选择NVIDIA控制面板:

a7271db588665bb2281a40bbcb881206.png

4、点击左下角的系统信息,出现如下所示:

6d4e8e0ae7dc4fba4d23b5d2712b5b15.png

5、点击组件,出现:

3960d060b4bcd330e5fa31c24e1e4acb.png

所选中区域的产品名称那列即为当前所支持的cuda的最高版本,可以往较低的版本进行安装

注意:不要安装cuda11.1版本,目前没有支持的pyTorch版本

上面显示的是cuda9.0,但是上面的图是我找的网上的图,在我实际安装过程中,显示的是cuda11.1,因此在安装过程中,还需要根据pyTorch版本的对应,因此我选择安装的cuda版本是10.0.

6、开始下载cuda

知乎 - 安全中心developer.nvidia.com

选择对应版本的进行安装既可,但是安装过程需要注意一步,就是有一个选项是自定义安装和推荐安装这两个选项,注意选择自定义安装,因为推荐安装是安装的全家桶。

903fcd92fe4843b181a7979507e7c2be.png

d53a8c9f955588ce4356525789a18e10.png

之后一步步安装既可,选择默认安装目录比较方便。

7、下面配置cuda的环境变量,找到系统属性中的环境变量,点击

9751c84dec7aeba8eb83ed1e74749b23.png

之后找到系统变量中的path,进行编辑-》新建-》确定操作,添加环境变量:

59569ca4d4482f03230b11da1e59910d.png

添加如下两条环境变量,是cuda的安装目录,因此选择默认的安装目录较好:

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0libx64

c5b187089026221d8f598c9ba4d22e28.png

然后,可以在终端中输入nvcc -V进行验证cuda是否安装成功:可以重启使得环境变量生效,我是没有重启。但是过了一会就生效了。显示效果如下:

baf72c170a663a1dffd74e71379c47dc.png

cuda即安装成功。

8、下面安装cudnn;

知乎 - 安全中心developer.nvidia.com

必须选择和cuda版本对应的cudnn版本号,我是cuda10.0,因此选择下图所示的版本

7af74f2c73147f541b618df56ad57e14.png

把下载的cudnn压缩包进行解压,在cudnn的文件夹下,把bin,include,lib文件夹下的内容对应拷贝到cuda相应的bin,include,lib下即可,最后安装完成。

9、安装pyTorch

进入pyTorch官网,可以看到有conda,pip等不同的命令行安装方式,可以选择自己方便的安装方式进行安装。

8fe6588c1baa83fc1ffbcc4284659ccb.png

同时,可以点击下列链接,安装自己需要的pyTorch版本。

4bcb9696fc0c2aa12fe7eb8c436fe164.png

至此,安装结束!

卸载

如果安装错了cuda,利用控制面板卸载即可,根据安装的日期逐个进行删除,最后把c盘program file下相应cuda troolkit的安装文件夹进行删除即可,即可完成删除。

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP