Linux下pytorch和cuda的版本对应及安装 Linux下pytorch和cuda的版本对应及

1. 根据驱动版本选择cuda版本

官网查询链接：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

文章图片

2. 根据cuda版本选择pytorch版本

进入pytorch官网：https://pytorch.org
选择不同的cuda版本会有不同的安装指令
需要安装更早的版本点击Previous versions of PyTorch根据需要查找安装指令
有时候用列出的指令安装总是报错，可以进入下面的链接查找对应版本，然后 wget + 文件地址下载，pip install + 文件名安装（直接pip install + 文件地址好像也可以安装）
（选择下载文件安装时要注意对应自己的python版本和操作系统）
（torch和torchvision的版本对应可以参考上面同一个指令里的版本对应）
（torchvision直接pip install torchvision==version就可以安装）

【Linux下pytorch和cuda的版本对应及安装】汇总的：https://download.pytorch.org/whl/torch_stable.html
CPU-only： https://download.pytorch.org/whl/cpu/torch_stable.html
CUDA 8.0：https://download.pytorch.org/whl/cu80/torch_stable.html
CUDA 9.0：https://download.pytorch.org/whl/cu90/torch_stable.html
CUDA 9.2：https://download.pytorch.org/whl/cu92/torch_stable.html
CUDA 10.0：https://download.pytorch.org/whl/cu100/torch_stable.html
3. pytorch和torchvision、torchaudio的版本对应

pytorch	torchvision	torchaudio
1.8.1	0.9.1	0.8.1
1.8.0	0.9.0	0.8.0
1.7.1	0.8.2	0.7.2
1.7.0	0.8.0	0.7.0
1.6.0	0.7.0
1.5.1	0.6.1
1.5.0	0.6.0
1.4.0	0.5.0
1.2.0	0.4.0
1.1.0	0.3.0
1.0.1	0.2.2
1.0.0	0.2.1
0.4.0	0.2.1