Pytorch官網:https://pytorch.org/
安裝的時候如果選擇在官網上下載安裝的話下載會很慢,試過梯子也是很慢,此處提供一種快速安裝的方法。
1.由於我是window系統,我選擇了國內清華源下載,https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/,找到自己適配的版本將其下載下來
注意:py3.7是指你的python版本是python3.7;
cuda100是指cuda的版本是10.0版本。
如果不知道自己的電腦的cuda版本號是多少,可以在cmd輸入以下命令查看:
nvcc -V
我的是python3.7,cuda是10.0,在清華源那里找的時候就找這個pytorch-1.0.1-py3.7_cuda100_cudnn7_1.tar.bz2。
2.下面是我用git命令行進行conda的安裝(根據自己習慣來,這里用管理員身份打開Aaconda Prompt執行該命令也是可以的。。)
(1)進入下載剛才的文件的目錄下
(2)執行以下命令:
conda install --use-local pytorch-1.0.1-py3.7_cuda100_cudnn7_1.tar.bz2
結果顯示:
備注:如果遇到權限問題,就用管理員身份打開Aaconda Prompt執行該命令。