原命令,例:
# CUDA 11.3
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
使用清华源,例:
# CUDA 11.3
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64
注意避坑(大坑)
清华源中如果有cpuonly这个包,会强制你装cpu版本的pytorch,因此需要卸载这个包再装执行上面的命令。
查看是否有这个包:
conda list
卸载cpuonly:
conda uninstall cpuonly