Pytorch 官網(wǎng)https://pytorch.org/
捕獲.PNG
選擇合適的版本,安裝默認(rèn)是官網(wǎng)海外服務(wù)器,添加國(guó)內(nèi)鏡像源就可以實(shí)現(xiàn)快速的download,依次執(zhí)行:
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
最后一行代碼和上面官網(wǎng)不同的是,去掉了-c pytorch,這樣才能換用國(guó)內(nèi)清華鏡像。然后大家要是用別的版本只要把cudatoolkit=10.1的10.1換掉就行。
清華鏡像官網(wǎng):https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
一般情況會(huì)在usr目錄,(win在C盤(pán),用戶,你的名字的目錄中有個(gè).condarc,Mac也在這個(gè)下面,比如我的在/Users/yuanhust/.condarc【如果mac中對(duì).開(kāi)頭文件隱藏,按住command+shift+.顯示隱藏】,Ubuntu也是),用記事本打開(kāi).condarc,把下面這個(gè)復(fù)制進(jìn)去也能修改:
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