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)