Pytorch 和CUDA版本对应关系

2023-05-09 23:52:17 浏览数 (1)

大家好，我是Peter~

本文记录下深度学习中Pytorch和cuda对应版本关系。

官方地址：https://pytorch.org/get-started/previous-versions/

查看CUDA版本

使用nvidia-smi命令显示的cuda版本信息

代码语言：python代码运行次数：0复制
nvidia-smi

需要注意的是：注意低版本的Pytorch是否向上支持更高版本的CUDA。

高版本的Pytorch一般能兼容低版本CUDA

Pytorch -V 1.0.0

代码语言：python代码运行次数：0复制
# CUDA 10.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch
 
# CUDA 9.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda90 -c pytorch
 
# CUDA 8.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch
 
# CPU Only
conda install pytorch-cpu==1.0.0 torchvision-cpu==0.2.1 cpuonly -c pytorch

Pytorch -V 1.2.0

代码语言：python代码运行次数：0复制
# CUDA 9.2
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch
 
# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
 
# CPU Only
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch

Pytorch -V 1.5.0

代码语言：python代码运行次数：0复制
# CUDA 9.2
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=9.2 -c pytorch
 
# CUDA 10.1
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorch
 
# CUDA 10.2
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 -c pytorch
 
# CPU Only
conda install pytorch==1.5.0 torchvision==0.6.0 cpuonly -c pytorch

Pytorch -V 1.7.0

代码语言：python代码运行次数：0复制
# CUDA 9.2
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=9.2 -c pytorch
 
# CUDA 10.1
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch
 
# CUDA 10.2
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch
 
# CUDA 11.0
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch
 
# CPU Only
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cpuonly -c pytorch

Pytorch -V 1.8.0

代码语言：python代码运行次数：0复制
# CUDA 10.2
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
 
# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
 
# CPU Only
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cpuonly -c pytorch

Pytorch -V 1.9.0

代码语言：python代码运行次数：0复制
# CUDA 10.2
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch
 
# CUDA 11.3
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.3 -c pytorch -c conda-forge
 
# CPU Only
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cpuonly -c pytorch

Pytorch -V 1.10.0

代码语言：python代码运行次数：0复制
# ROCM 4.2 (Linux only)
pip install torch==1.10.0 rocm4.2 torchvision==0.11.0 rocm4.2 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
 
# ROCM 4.1 (Linux only)
pip install torch==1.10.0 rocm4.1 torchvision==0.11.0 rocm4.1 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
 
# ROCM 4.0.1 (Linux only)
pip install torch==1.10.0 rocm4.0.1 torchvision==0.10.1 rocm4.0.1 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
 
# CUDA 11.1
pip install torch==1.10.0 cu111 torchvision==0.11.0 cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
 
# CUDA 10.2
pip install torch==1.10.0 cu102 torchvision==0.11.0 cu102 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
 
# CPU only
pip install torch==1.10.0 cpu torchvision==0.11.0 cpu torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
深度学习 pytorch 人工智能 机器学习机器学习 深度学习 pytorch

0 人点赞