首頁 > 軟體

Ubuntu 17.04 安裝 pytorch and pytorchvision

2020-06-16 17:15:00

Ubuntu 17.04 安裝 pytorch and pytorchvision (conda,pip 均失敗後採用原始碼安裝)。

最近在研究pytorch,安裝的時候遇到不少坑,在此做下記錄。

pytorch是torch移植到Python上的版本,作為python3的第三方庫,我們首先想到的就是利用pip來進行安裝,但是直接使用

pip3 install pytorch  (因為我使用的是python3,因此此處為pip3),會報如下錯誤


 

提示我們要去pytorch的官網進行下載,開啟官網,可以按照安裝方式(pip ,conda,source)和python版本,cuda版本來進行安裝,由於本國特殊的網路國情,一般是打不開,或者是無法檢視 run the command 之後的指令的。


 

在題主進行科學上網以後,得到了利用pip進行安裝的相關的指令,(該指令為linux,pip,python3.6,none-cuda)

pip install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl    (linux下,即使cuda選擇none,還是預設安裝cuda75版本)

pip install torchvision

卻發現torchvision已經不再支援pip,pytorch可以正常使用


 

看提示,建議我們選擇miniconda進行安裝,於是乎轉戰conda,由於我們學校提供了官方的映象庫,所以我選擇直接安裝了anaconda,如何在ubuntu上安裝anaconda有很多文章可以參考,可以之前前往  https://www.continuum.io/downloads/  進行安裝。

利用conda進行安裝的指令如下:

cuda7.5 

conda install pytorch torchvision -c soumith

cuda8.0

conda install pytorch torchvision cuda80 -c soumith

然而!利用conda進行安裝時會報condaHttpError錯誤,起初我認為是我國特殊的網路環境造成的,因此,將conda的源切換為我們學校的映象源,仍然會報HttoError,但是安裝別的第三方庫可以成功。此時我是絕望的=。=這2個第三方包管理器太不靠譜,我在starkoverflow和github上找了一下,有很多人碰到了類似的問題,也有一些回答,但經過我的嘗試,沒有一個能解決我的問題。


 

最終解決方法----原始碼編譯

原始碼包在github上的託管地址為

https://github.com/pytorch/vision
https://github.com/pytorch/pytorch

分別克隆2個包的原始碼進行安裝:

git clone git@github.com:pytorch/pytorch.git

git clone git@github.com:pytorch/vision.git

然後,cd到克隆的目錄下

cd pytorch

python3 setup.py install

cd vision

python3 setup.py install

注意,若安裝時提示找不到cmake,直接sudo apt install cmake 即可

安裝完成之後,進入ipython3,檢視是否安裝成功(直接在python命令列檢視也可)

 


 

之後就可以暢遊pytorch和pytorchvision。


IT145.com E-mail:sddin#qq.com