conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.1

这里需要注意:

-c pytorch
  • 需要去掉,因为这个参数指明了官网的优先级大于清华的镜像源。
  • cudatoolkit的版本号根据自己的cuda版本添加。

另外,如需采用pip install来安装所需要的库,临时的方法是在pip install后面添加指令:

-i https://pypi.tuna.tsinghua.edu.cn/simple

来指定channel。

例如:

pip3 install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple

这里采用的是清华的源,其他的源包括:

1.豆瓣:http://pypi.douban.com/simple
2.中国科学技术大学 : https://pypi.mirrors.ustc.edu.cn/simple
如需全局修改,请参考 这里

GitHub 加速计划 / vi / vision
15.85 K
6.89 K
下载
pytorch/vision: 一个基于 PyTorch 的计算机视觉库,提供了各种计算机视觉算法和工具,适合用于实现计算机视觉应用程序。
最近提交(Master分支:2 个月前 )
868a3b42 5 天前
e9a32135 14 天前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐