离线安装torch和torchvision

一、查看Windows11的cuda版本

打开NVIDIA control panel应用,点击左下角的系统信息
离线安装torch和torchvision_第1张图片
进入后点击组件,再查看cuda
离线安装torch和torchvision_第2张图片
ps:不知道需不需要查看cudnn,我没查看,网上的方法都没用,但是不影响我安装成功。

二、下载对应版本的torch并离线安装

进入任意一个可用的镜像源下载你的python版本和cuda版本对应的torch
我用的是清华镜像源
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

我的python是3.8 cuda11.6,这里我选择的是下面这个包下载
pytorch
torchvison
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

你可能感兴趣的:(环境配置,pytorch,深度学习,python)