【工具】PyTorch安装（匹配CUDA版本）

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