本文主要是介绍conda安装pytorch使用清华源,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
原命令,例:
# 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
参考链接
清华源conda 安装gpu版本的pytorch总是推送cpu版本解决办法
这篇关于conda安装pytorch使用清华源的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!