centos安裝torch==1.4.0與相關細節

2022 年 7 月 31 日
筆記

對於某些直接安裝torch==1.4.0報錯的情況（沒錯，就是我遇到了）

在網上查找了，大概的解決方法是先安裝一個低版本的torch和torchvision, torchvision是pytorch中專門用來處理影像的庫，然後再安裝torch==1.4.0。實踐過後的操作如下。

實操

首先執行

pip install torch==1.3.1 -f //download.pytorch.org/whl/torch_stable.html
pip install torchvision==0.4.1 -i //pypi.tuna.tsinghua.edu.cn/simple

安裝上低版本的torch後，就可以進行torch==1.4.0的安裝

網上給出的安裝命令是

pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f //download.pytorch.org/whl/torch_stable.html -i //pypi.tuna.tsinghua.edu.cn

但是，我在實際操作過程中報錯了，可能是清華的鏡像源中刪除了這個版本的cpu的torch的安裝。

所以，我的實際的安裝命令沒有採用鏡像源，或者你也可以換成其他的鏡像源進行嘗試。

CPU版本

pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f //download.pytorch.org/whl/torch_stable.html
GPU版本

不指定cuda的版本，其會自動匹配一個cuda的版本進行安裝，我這就是匹配到了cuda9.2的版本。

pip install torch==1.4.0 torchvision==0.5.0 -f //download.pytorch.org/whl/torch_stable.html

也可以自己指定cuda的版本,比如安裝cuda10.0的版本，注意有些cuda的版本不一定可以適配torch==1.4.0，這個要自己去查找匹配。

pip install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f //download.pytorch.org/whl/torch_stable.html

查看安裝的torch版本

pip list 或者 conda list 都可以查看到torch的版本號

注意：

如果指定從官方的下載地址 : //download.pytorch.org/whl/torch_stable.html 下載pytorch 不管指不指定torch 和 torchvision 的版本，下載GPU版本，利用pip list 查看 torch的版本的時候版本的後面會有一個cuda版本的後綴。

例如 pip install torch==1.4.0 torchvision==0.5.0 -f //download.pytorch.org/whl/torch_stable.html 安裝時，雖然官方源中有torch=1.4.0 這個版本，但是其找到的還是 torch=1.4.0+cu92這個版本，並下載了這個版本。