CUDA Toolkit Archive(含PyTorch所有版本)
根据自己需要的CUDA版本和PyTorch版本找到对应的下载命令,这里以PyTorch1.10.1 CUDA 11.3 为例。
Linux and Windows
# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch
# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
# CPU Only
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cpuonly -c pytorch