您的位置:首页 > 科技 > IT业 > conda安装pytorch使用清华源

conda安装pytorch使用清华源

2024/12/21 22:08:49 来源:https://blog.csdn.net/zwx54088/article/details/139702321  浏览:    关键词: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版本解决办法

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com