PyTorch0.4.1 GPU版本安装完整步骤(Anaconda3 + Python3.6 + gpu版本,含CUDA9.2安装、环境配置)

发布时间:2022-08-19 12:34

目录

一、下载并安装cuda9.2

1. 首先查看电脑显卡cuda版本,看最高支持的cuda为多少

2. 下载cuda9.2安装包

3. 安装cuda9.2

二、下载并安装cudnn (for cuda9.2)

1. 下载cudnn

2. 安装cudnn

三、配置环境变量

四、安装GPU版本的PyTorch0.4.1

1. 准备:

2. 下载gpu版本的torch、torchvision的whl包(cu92):

3. 离线安装gpu版本的torch、torchvision(cu92):


亲测有用!!!

本人电脑:win10 + 有英伟达显卡(GeForce GTX 1650) + Anaconda3 5.2.0 (其中Anaconda里的Python版本是3.6),要安装CUDA、PyTorch 0.4.1 版本(gpu版本)

目标:安装gpu版本的PyTorch0.4.1(cuda9.2+pytorch0.4.1)

写在前面:由于最近有需要,要安装pytorch比较老的版本(零点几的版本),恰好本人电脑带有显卡,所以想要直接安装个gpu版本的pytorch古早版本。pytorch版本老,那么相应的那个pytorch版本支持的cuda版本也比较老,所以我接下来就打算:1)先装个cuda9.2(大致有cuda8.0、cuda9.0、cuda9.1、cuda9.2、cuda10.0等等很多版本),并安装好对应的cudnn,如果到时候cuda9.2太老的话,可以再换装其他版本。。。。。2)安装好cuda9.2和cudnn9.2之后,然后再安装一个gpu版本(cu92,和前面的cuda9.2对应)的PyTorch0.4.1版本。(本来打算安装PyTorch0.4.0之前的版本,但一是以前版本的安装包还挺少的,二是以前版本对应的gpu版本下匹配的cuda版本也很少很老,选择的余地比较少。所以就先打算安装cu92/PyTorch0.4.1试试水,不行的话可以再换装其他。)

一、下载并安装cuda9.2

1. 首先查看电脑显卡cuda版本,看最高支持的cuda为多少

在命令行通过命令nvidia-smi查看cuda版本,如下图。可以看到我的显卡最高支持cuda11.2,即可以装cuda10.0、cuda11.0等,但不能装cuda11.3及以上。(查看显卡和cuda版本可以参考之前博客:win10查看Nvidia显卡、cuda版本)

PyTorch0.4.1 GPU版本安装完整步骤(Anaconda3 + Python3.6 + gpu版本,含CUDA9.2安装、环境配置)_第1张图片

2. 下载cuda9.2安装包

cuda官网下载地址:CUDA Toolkit Archive | NVIDIA Developer

点击上面的官网下载链接,选择cuda9.2,点进去。(此处选择下载cuda9.2,是一个大约1.5G的exe文件。)

  • 也可以使用百度云下载cuda9.2和对应的cudnn9.2,百度云下载链接: CUDA9.2和配套的cuDNN百度云直达下载,提取码:81v4。来自博客(pytorch gpu版本的最全安装教程,含环境配置、CUDA(百度云下载)安装程序失败解决方案)

ItVuer - 免责声明 - 关于我们 - 联系我们

本网站信息来源于互联网,如有侵权请联系:561261067@qq.com

桂ICP备16001015号