由于TensorFlow的靜態(tài)圖特點(diǎn)導(dǎo)致最近的一些實(shí)驗(yàn)無法完成，決定轉(zhuǎn)戰(zhàn)pytorch，自己做一個安裝記錄，也希望對有需要的小伙伴有所幫助

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

在cmd中直接安裝即可

有的小伙伴（包括我）會發(fā)現(xiàn)直接使用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 <本地下載地址>即可

接下來是測試是否安裝成功的環(huán)節(jié)

import torch

以及

import torchvision

在這步中，可捕獲錯誤

ImportError: No module named torchvision

這里通過查找資料可以發(fā)現(xiàn)與cuda8.0+python3.6+pytorch1.0對應(yīng)的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

可以在提示信息中發(fā)現(xiàn)有0.4.1這個版本，再次嘗試

pip3 install torchvision==0.4.1

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

再次測試

好的，開啟pytorch大門！

?

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。