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

pip install torchvision

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

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

本人嘗試了0.1.6版本（pip install torchvision == 0.1.6），發(fā)現(xiàn)版本過低，在使用torchvision.transforms時報錯：

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

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

大家如果有什么問題可以繼續(xù)調(diào)試版本，查閱torch和torchvision的版本對應(yīng)關(guān)系，二者版本不匹配有時也會報錯！

總結(jié)

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

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