最近兩天被windows下pytorch的安裝給搞得很煩了,不過在今天終於安裝成功了,如下圖所示
下面詳細說下此次安裝的詳細記錄吧。我的電腦環境是Windows10+cuda9.0+cudnn7.1。
首先按照流程百度安裝教程,然后發現大部分都是用官網https://pytorch.org/提供的命令安裝,如下圖所示
然后再安裝的過程中,出現了很多HTTP之類的錯誤,而根據我安裝其他python庫的經驗的話,意思就是訪問國外的網站的話速度太慢了直接導致斷掉,然后我根據錯誤來查找解決方案,得到的結果很多都是在.condarc文件下添加清華源。如下圖類似。
然后繼續運行上面的命令進行安裝,發現毫無用處,還是報之前的錯誤。
最終解決方案:就是離線安裝,在清華源鏡像網站https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/下找到torch和torchvision的安裝文件。以我的環境為例,就是下面兩個圖所示,版本的話看自己的需求了。
然后依次運行下面這兩個命令
conda install --offline pytorch-1.1.0-py3.6_cuda90_cudnn7_1.tar.bz2
conda install --offline torchvision-0.3.0-py36_cu90_1.tar.bz2
用完這兩個安裝包就可以刪除了哈哈哈,上面命令中安裝的路徑可以自定義,因為我把這兩個包放在了根目錄,為了避免浪費C盤空間,所以用完就刪掉了。
如此就成功安裝PyTorch了。