Ubuntu离线安装PyTorch-GPU

Ubuntu离线安装PyTorch-GPU

之所以采用离线安装，是因为PyTorch实在是太大了，即使换源在线安装也很难成功。

一.环境简介

Ubuntu20.04+nvidia-driver-510 + CUDA11.6 + cuDNN v8.4.0+Python3.10

二.下载并安装

step1：根据CUDA版本（11.6）和Python版本(3.10)选择PyTorch版本

进入PyTorch下载官网

文件名对应PyTorch的版本号，cu116表示对应CUDA11.6的GPU版本，cp310表示对应Python3.10版本，且又是Ubuntu里安装，综上选择相应的1.13.0版本的PyTorch，因此下载

cu116/torch-1.13.0%2Bcu116-cp310-cp310-linux_x86_64.whl

step2：根据PyTorch版本选择torchvision版本

可通过该网站中获取版本关系：PyTorch与torchvision版本关系对照网站

经比对1.13.0版本的PyTorch，对应的是0.14.0版本的torchvision，故下载

cu116/torchvision-0.14.0%2Bcu116-cp310-cp310-linux_x86_64.whl

step3：创建并激活conda环境

conda create -n mypytorch python=3.10
conda activate mypytorch

step4：离线安装

在刚刚创建的conda虚拟环境mypytorch下，终端输入以下指令（文件名注意替换）完成安装。

cd ./下载  #切换到下载好文件的路径
pip install torch-1.13.0+cu116-cp310-cp310-linux_x86_64.whl
pip install torchvision-0.14.0+cu116-cp310-cp310-linux_x86_64.whl

三.验证安装

输入以下指令进行验证

import torch
torch.cuda.is_available()  ## 输出应该是True
torch.__version__