一、查看Windows11的cuda版本
打开NVIDIA control panel应用,点击左下角的系统信息
进入后点击组件,再查看cuda
ps:不知道需不需要查看cudnn,我没查看,网上的方法都没用,但是不影响我安装成功。
二、下载对应版本的torch并离线安装
进入任意一个可用的镜像源下载你的python版本和cuda版本对应的torch
我用的是清华镜像源
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
我的python是3.8 cuda11.6,这里我选择的是下面这个包下载
ps:这里需要注意版本对应,torch,torchvison的版本不要出错。可以去PyTorch中torch、torchvision、torchaudio版本对应关系这篇文章中查看。还可以去官方文档1、官方文档2中查看。
下载完成后,打开cmd,进入环境,并cd到文件夹中输如下面代码即可
conda install --offline pytorch-1.12.0-py3.8_cuda11.6_cudnn8_0.tar.bz2
conda install --offline torchvision-0.13.0-py38_cu116.tar.bz2
出现下面两个情况即为安装成功
(base) PS C:\Users\19192> conda install --offline pytorch-1.12.0-py3.8_cuda11.6_cudnn8_0.tar.bz2
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
(base) PS C:\Users\19192> conda install --offline torchvision-0.13.0-py38_cu116.tar.bz2
Downloading and Extracting Packages
######################################################################################## | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done