深度學習實踐操作—從小白到大白（八）：安裝Pytorch0.3到特定的Anaconda環境

---

深度學習實踐操作—從小白到大白

目錄

---

八. 安裝Pytorch到特定的Anaconda環境

參考：官方網站https://github.com/pytorch/pytorch#from-source

教程中會將Pytorch安裝到Anaconda下一個單獨的環境中，因此需要事先安裝Anaconda。在實際使用Linux系統時，每次接觸一個新的東西的時候，利用Anaconda創建一個新環境，並且所需要的依賴項都儘可能在這個單獨環境中進行操作，是一個很好的習慣。

還需要安裝CUDA和cuDNN，CUDA版本至少7.5，cuDNN版本至少6.x

CUDA、cuDNN以及Anaconda的安裝請點擊目錄有對應教程。

1. 配置環境變量

gedit ~/.bashrc  
export CMAKE_PREFIX_PATH=/home/ggy/anaconda2/bin  #改爲Anaconda安裝目錄

2. 創建新環境MyPytorch

conda create -n MyPytorch #MyPytorch是爲環境自定義的名字

3. 安裝依賴包（需要聯網）

conda install numpy pyyaml mkl setuptools cmake gcc cffi
或者
conda install -n MyPytorch numpy pyyaml mkl setuptools cmake gcc cffi 安裝到環境MyPytorch

4. 加入GPU支持（需要聯網）

conda install -c soumith magma-cuda80 #如果是CUDA 7.5 ，magma-cuda75 
#或者
conda install -n MyPytorch -c soumith magma-cuda80 #安裝到環境MyPytorch

安裝這個的時候，國內由於網絡原因可能安裝不成功，可以使用另外一個源：

conda install -c pytorch magma-cuda80 
#或者
conda install -n MyPytorch -c pytorch magma-cuda80#安裝到環境MyPytorch

如果也不行可以下載安裝文件在conda中離線安裝：

https://anaconda.org/pytorch/magma-cuda80/2.3.0/download/linux-64/magma-cuda80-2.3.0-1.tar.bz2

source activate MyPytorch #激活想要安裝的環境
cd /home/ggy/anaconda2/envs/MyPytorch #進入到環境目錄下
conda install magma-cuda80-2.3.0-1.tar.bz2

5. 安裝Pytorch（需要聯網）

如果安裝到特定目錄，運行下兩條命令：

source activate MyPytorch #激活想要安裝的環境
cd /home/ggy/anaconda2/envs/MyPytorch #進入到環境目錄下

安裝

git clone --recursive https://github.com/pytorch/pytorch #下載
cd ./pytorch
python setup.py install #安裝

6. 安裝torchvision（需要聯網）

如果安裝到特定目錄，運行下兩條命令：

source activate MyPytorch #激活想要安裝的環境
cd /home/ggy/anaconda2/envs/MyPytorch #進入到環境目錄下

安裝

git clone --recursive https://github.com/pytorch/vision #下載
cd ./vision
python setup.py install #安裝