- 用conda安裝Pytorch過程中會連接失敗,這是因爲Anaconda.org的服務器在國外,需要切換到國內鏡像源:
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- 設置搜索時顯示通道地址
conda config --set show_channel_urls yes
- 創建虛擬環境並激活之,在該環境下安裝下面的包:
conda create pytorch python=3.6
- 安裝pytorch
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0