发布时间: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试试水,不行的话可以再换装其他。)
在命令行通过命令nvidia-smi查看cuda版本,如下图。可以看到我的显卡最高支持cuda11.2,即可以装cuda10.0、cuda11.0等,但不能装cuda11.3及以上。(查看显卡和cuda版本可以参考之前博客:win10查看Nvidia显卡、cuda版本)
cuda官网下载地址:CUDA Toolkit Archive | NVIDIA Developer
点击上面的官网下载链接,选择cuda9.2,点进去。(此处选择下载cuda9.2,是一个大约1.5G的exe文件。)