Conda 常用命令与 PyTorch-CUDA 兼容速查

环境管理 · 包管理 · 查询 · 配置 · CUDA/PyTorch 兼容

Conda 常用命令

PyTorch 与 CUDA 兼容速查

PyTorch版本 CUDA版本 安装命令(conda)
2.2 12.1 conda install pytorch=2.2 torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
2.1 11.8 conda install pytorch=2.1 torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
2.0 11.8 conda install pytorch=2.0 torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
1.13 11.7 conda install pytorch=1.13 torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
1.12 11.3 conda install pytorch=1.12 torchvision torchaudio cudatoolkit=11.3 -c pytorch
1.11 11.3 conda install pytorch=1.11 torchvision torchaudio cudatoolkit=11.3 -c pytorch
1.10 10.2 conda install pytorch=1.10 torchvision torchaudio cudatoolkit=10.2 -c pytorch
1.9 11.1 conda install pytorch=1.9 torchvision torchaudio cudatoolkit=11.1 -c pytorch
1.8 11.1 conda install pytorch=1.8 torchvision torchaudio cudatoolkit=11.1 -c pytorch
1.7 10.2 conda install pytorch=1.7 torchvision torchaudio cudatoolkit=10.2 -c pytorch
1.6 10.1 conda install pytorch=1.6 torchvision torchaudio cudatoolkit=10.1 -c pytorch
1.5 10.1 conda install pytorch=1.5 torchvision torchaudio cudatoolkit=10.1 -c pytorch
官方兼容表: PyTorch 官网版本兼容查询
常用 CUDA 安装命令:
# CUDA 12.1
conda install nvidia/label/cuda-12.1.0::cudatoolkit
# CUDA 11.8
conda install cudatoolkit=11.8
# CUDA 11.7
conda install cudatoolkit=11.7
# CUDA 11.3
conda install cudatoolkit=11.3
# CUDA 10.2
conda install cudatoolkit=10.2

Pip 常用命令与国内源