天天看點

Linux安裝pytorch的具體過程以及其中出現問題的解決辦法

Linux安裝pytorch的具體過程以及其中出現問題的解決辦法

1.安裝Anaconda

  安裝步驟參考了官網的說明:https://docs.anaconda.com/anaconda/install/linux.html

  具體步驟如下:

     首先,在官網下載下傳位址 https://www.anaconda.com/download/下載下傳linux版本,這裡選用python 3.6版本的anaconda.

     然後, 将下載下傳好的Anaconda3-4.4.0-Linux-x86_64.sh放到/usr/tiny目錄下,并進入該目錄

     在目前目錄下用bash指令安裝anaconda

root bash    Anaconda3-4.4.0-Linux-x86_64.sh       
   安裝完畢後,輸入如下指令使得配置生效      

source ~/.bashrc

  為驗證安裝正确性,可以通過如下指令進行測試

conda list
      

2.用conda安裝/更新pytorch的依賴庫

首先,激活Anaconda配置檔案

source ~/.bashrc
      

使用conda install指令可以友善快捷得安裝或者更新所需的内容

conda install numpy pyyaml mkl setuptools cmake gcc cffi
      

 

3.安裝pytorch

本文對pytorch官網給出的三種安裝方式分别進行了測試,結果如下:

首先,使用conda進行安裝,指令如下,其中cuda80可以根據cuba版本調整為cuda75或者直接省略

conda install pytorch torchvision cuda80 -c soumith
      

因為下載下傳速度隻有幾KB,忍無可忍直接終止了程序。

然後直接從官網下載下傳了source檔案

git clone --recursive https://github.com/pytorch/pytorch      

然後用setup.py進行安裝

python setup.py install
      

安裝過程顯示報錯,缺失python.h檔案,于是把python-dev和python3-dev一起裝了一遍 

sudo apt-get install python-dev       
sudo apt-get install python3-dev       

再次運作 

python setup.py install
      

顯示把torch安裝到了python2.7的庫檔案夾裡,本來想删掉python-dev再安裝一遍pytorch,沒有找到相應的辦法,遂考慮通過pip進行安裝

通過pip進行pytorch安裝的過程如下:

pip install http://download.pytorch.org/whl/cu75/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl  
pip install torchvision    
      

下載下傳速度在幾MB左右,安裝過程十分迅速,且沒有報錯。

安裝完畢之後,激活Anaconda配置檔案

source ~/.bashrc
 
      

進入python指令行模式:

python
      

測試安裝是否正确:

from __future__ import print_function
import torch
a = torch.ones(5)
print(a)      

繼續閱讀