Pytorch 官网https://pytorch.org/
选择合适的版本,安装默认是官网海外服务器,添加国内镜像源就可以实现快速的download,依次执行:
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 install pytorch torchvision cudatoolkit=10.1
最后一行代码和上面官网不同的是,去掉了-c pytorch,这样才能换用国内清华镜像。然后大家要是用别的版本只要把cudatoolkit=10.1的10.1换掉就行。
清华镜像官网:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
一般情况会在usr目录,(win在C盘,用户,你的名字的目录中有个.condarc,Mac也在这个下面,比如我的在/Users/yuanhust/.condarc【如果mac中对.开头文件隐藏,按住command+shift+.显示隐藏】,Ubuntu也是),用记事本打开.condarc,把下面这个复制进去也能修改:
channels:
- defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud