1. 程式人生 > >Ubutnu16.04安裝pytorch

Ubutnu16.04安裝pytorch

code 4.4 nbsp 錯誤 detail cond edit family 3.6

1、下載Anaconda3

首先需要去Anaconda官網下載最新版本Anaconda3(https://www.continuum.io/downloads),我下載是是帶有python3.6的Anaconda3-4.4.0-Linux-x86_64.sh。

2、安裝Annconda3

bash Anaconda3-4.4.0-Linux-x86_64.sh 在home/ubuntu出現anaconda3文件夾(註:ubuntu是系統用戶名、下同)。

source ~/.bashrc

3、配置環境變量

gedit ~/.bashrc 加入下面那行命令

export CMAKE_PREFIX_PATH=/home/ubuntu/anaconda3/bin

gedit ~/.bash_profile 加入下面那行命令 export PATH="/home/ubuntu/anaconda3/bin:$PATH"

4、還需要安裝一些依賴庫

conda install numpy mkl setuptools cmake gcc cffi conda install -c soumith magma-cuda75 # or magma-cuda80 if CUDA 8.0 (若出現conda not found之類的錯誤,運行命令source ~/.bashrc 、然後在執行)

5、源碼安裝pytorch

(1)可以用github上的這個鏈接(https://github.com/pytorch/pytorch)的源碼,git cone(首先要建立git與github的ssh)到本地,運行 pip install -r requirements.txt 、python setup.py install安裝即可。 註:不推薦使用這個github上面的源碼安裝、因為可能是因為pytorch版本的問題,導致我後面使用Tensor張量的時候出現了一下問題,例如經過 Temsor.max()函數之後把我的二維的變成了一維。 (2)http://pytorch.org/ ,conda命令下載安裝。

6、測試

安裝結束之後,打開python輸入import torch 若不出現錯誤就已經安裝成功了。

7、自己遇到的問題

(1)若心打開一個terminal在python中輸入import torch出現不能發現torch moudle的錯誤,很疑惑,明明pytorch已經安裝成功了,為什麽還會出現這種錯誤呢,這個問題應該是環境變量沒有配置好的問題,可以自己上網查看,不嫌麻煩的話也可以運行命令 source ~/.bash_profile 在運行import torch就可以了,但還是要註意安裝的過程中就把環境變量配置好。 (2)安裝過程中可能需要pip git vim等命令、有時候需要update的,要註意一下。

8、參考

http://blog.csdn.net/u013066730/article/details/62217401

Ubutnu16.04安裝pytorch