本人在工作時遇到了一個比較尷尬的問題，就是虛擬環境中已有固定版本的torch，但沒有安裝torchvision，因此很本能的安裝：

pip install torchvision

如果直接這樣安裝的話，我們發現會連帶安裝新版本的torch，由于不是本人的虛擬環境，不想改變別人的torch版本。因此該方法行不通。

就在思考如何分開安裝torch 和 torchvision，后來發現，只需下載較低版本的torchvision就可以解決我的問題。只要下載的版本大于0.3.x，就會下載更新torch。因此選擇安裝低版本的。

本人嘗試了0.1.6版本（pip install torchvision == 0.1.6），發現版本過低，在使用torchvision.transforms時報錯：

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? module 'torchvision.transforms' has no attribute 'xxxx'。

這時我們要適當調高版本，因此本人下載了0.2.1（pip install torchvision == 0.2.1），成功解決問題！

大家如果有什么問題可以繼續調試版本，查閱torch和torchvision的版本對應關系，二者版本不匹配有時也會報錯！

總結

以上是生活随笔為你收集整理的pytorch 之 分开安装 torch 和 torchvision的全部內容，希望文章能夠幫你解決所遇到的問題。

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