Lubuntu安裝Pytorch

PyTorch是啥

PyTorch官方對於PyTorch的定位爲：

• 一個使用GPU加速的numpy替換庫
• 一個深度學習研究平臺，提升最大靈活度和速度

具體點來說， PyTorch是一個Python包，是Torch在Python上的衍生，原先的Torch是用Lua語言寫的，雖然效率高，可是普及度不夠，社區不夠大，改爲Python後，受衆範圍普遍了許多。而且有FaceBook這樣的巨頭支持，具備很好的發展前景。

轉載請註明出處：http://www.javashuo.com/article/p-bvvrgurb-dx.html

安裝

因爲Lubuntu是基於Ubuntu衍生而來，所以安裝過程能夠參照Ubuntu下安裝Pytorch的過程，通常有三種方式：基於conda安裝；基於pip文件安裝；基於源碼編譯。

基於conda安裝

1. 安裝conda命令(Anaconda或者Miniconda)

• Anaconda包括包管理器和不少經常使用package,所以若是不少庫你都尚未安裝,直接安裝Anaconda是不錯的辦法.

• 反之,若是不少依賴package你都安裝好了,那麼你安裝miniconda 就好,也就是僅僅安裝一個package管理器.

一般Anaconda的安裝包都是.sh格式,假設是filename.sh,那麼安裝是在前面採用sh命令就能夠進行安裝了

sh filename.sh

2.修改conda鏡像源

在終端中運行如下命令修改鏡像源,能夠明顯加速安裝.

# 優先使用清華conda鏡像
conda config --prepend channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

# 也可選用科大conda鏡像
conda config --prepend channels http://mirrors.ustc.edu.cn/anaconda/pkgs/free/

要查看鏡像源是否安裝成功的話,建議終端中運行如下命令:

conda config --set show_channel_urls yes

會生成一個~/.condarc文件,運行cat命令查看文件內容

cat ~/.condarc

顯示內容爲：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.ustc.edu.cn/anaconda/pkgs/free/
  - defaults
show_channel_urls: true

3. 安裝Pytorch

我安裝的方法是conda+python3.6+CPU,因此安裝命令以下:

conda install pytorch-cpu torchvision -c pytorch

基於pip文件安裝

1. 配置鏡像源

請參考博文Ubuntu16.10下配置pip國內鏡像源加速安裝. 
這一步是爲加速安裝torch vision進行的.

2. 下載.whl安裝文件

(1)速度慢可是版本最全： 
能夠按pytorch安裝指南的網址，如： 
http://download.pytorch.org/whl/cpu/torch-0.3.1-cp35-cp35m-linux_x86_64.whl直接下載.whl文件，具體網址請在pytorch官網查詢． 
(2)速度快可是版本可能不全： 
從pypi官網 https://pypi.python.org/packages下載須要的.whl文件 
也能夠從如下的pypi國內的鏡像源網站中下載相應的.whl文件: 
豆瓣(douban) http://pypi.douban.com/simple/ (推薦) 
阿里雲 http://mirrors.aliyun.com/pypi/simple/ 
中國科技大學 https://pypi.mirrors.ustc.edu.cn/simple/ 
清華大學 https://pypi.tuna.tsinghua.edu.cn/simple/ 
中國科學技術大學 http://pypi.mirrors.ustc.edu.cn/simple/ 

3. 安裝pytorch

用pip install .whl文件便可

pip install torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl
pip install torchvision

基於源碼編譯

pytorch在github上的地址

1.首先安裝各類依賴

export CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" # [anaconda root directory]

# Install basic dependencies
conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
conda install -c mingfeima mkldnn

# Add LAPACK support for the GPU
conda install -c pytorch magma-cuda80 # or magma-cuda90 if CUDA 9

2.git代碼

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

3. 安裝pytorch

python setup.py install

整個安裝過程會很漫長，請耐心等待。。。

 

測試

在終端中按如下順序運行命令:

Python
import torch 

 

 

 

參考連接：

http://www.javashuo.com/article/p-hnxfvelv-cq.html

 https://github.com/pytorch/pytorch#binaries