1. 程式人生 > >[ Pytorch ] ——安裝:使用Anaconda安裝Pytorch

[ Pytorch ] ——安裝:使用Anaconda安裝Pytorch

1、新增清華源,然後使用conda進行安裝。(注:0.3.1 及以後)

### for 0.4.0 and later
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
(注: 新增的映象源可以在 ~/home/你的系統名稱/.condarc 檔案中檢視)

# for CPU only packages
conda install pytorch-cpu torchvision-cpu

2、GPU版本

(你的環境)>> pip install torch torchvision

[參考文獻]