由于TensorFlow的靜態圖特點導致最近的一些實驗無法完成，決定轉戰pytorch，自己做一個安裝記錄，也希望對有需要的小伙伴有所幫助

先前安裝的是cuda8.0，并非最新版本，因此需要安裝對應版本的pytorch。在pytorch官網上https://pytorch.org/可以找到最新版本以及之前的版本

在cmd中直接安裝即可

有的小伙伴（包括我）會發現直接使用pip安裝基本不動，使用鏡像也不動（鏡像可參考https://blog.csdn.net/xo3ylAF9kGs/article/details/104104041/）

這里有一個比較好的解決方法是直接下載輪子，cuda8.0+python3.6可以使用https://download.pytorch.org/whl/cu80/torch-1.0.0-cp36-cp36m-win_amd64.whl，更多版本的下載地址可參考https://blog.csdn.net/qq_27825451/article/details/90033909

更多輪子https://download.pytorch.org/whl/cu100/torch_stable.html

下載到本地后再安裝簡直不要太快

使用pip install <本地下載地址>即可

接下來是測試是否安裝成功的環節

import torch

以及

import torchvision

在這步中，可捕獲錯誤

ImportError: No module named torchvision

這里通過查找資料可以發現與cuda8.0+python3.6+pytorch1.0對應的torchvision版本為0.4

然后安裝

pip install torchvision

可以得到如下四種錯誤

ERROR: Could not find a version that satisfies the requirement torch==1.4.0 (from torchvision) (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2)ERROR: No matching distribution found for torch==1.4.0 (from torchvision)ERROR: Could not find a version that satisfies the requirement torchvision==0.4 (from versions: 0.1.6, 0.1.7, 0.1.8, 0.1.9, 0.2.0, 0.2.1, 0.2.2, 0.2.2.post2, 0.2.2.post3, 0.3.0, 0.4.1, 0.5.0)ERROR: No matching distribution found for torchvision==0.4

可以在提示信息中發現有0.4.1這個版本，再次嘗試

pip3 install torchvision==0.4.1

這里由于某種未知力量，使用了pip3

再次測試

好的，開啟pytorch大門！

?

總結

以上是生活随笔為你收集整理的pytorch安装实录（win10+cuda8+pycharm+anaconda）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。