1. 程式人生 > >Linux+Anaconda+PyTorch 環境搭建

Linux+Anaconda+PyTorch 環境搭建

Outline

Installation Steps

Python

首先看一下Linux下的預設Python版本: python --version
2.7, 3+ 在安裝Anaconda時是有區別的。

Anaconda

為啥推薦用Anaconda呢,原因就是方便,下載速度也快點,而且可以建立虛擬環境,為不同的專案搭建不同的環境。pytorch官網下載的比較慢,有時會直接斷了。Anaconda是python科學計算的整合。
下載地址:

https://www.anaconda.com/download

  1. 根據Python的版本來選擇安裝檔案下載
  2. 把sh檔案放到你想安裝的資料夾內,開啟終端跳轉到該資料夾,執行命令:bash name.sh
  3. 執行過程中會給出安裝路徑:eg: Prefix=/home/your_pcname/anaconda_file/,如果想改的話,執行到此按照提示修改
  4. 顯示success後還沒完,執行sudo gedit /etc/profile,開啟profile檔案
  5. 在檔案末尾新增一行:export PATH=/home/your_pcname/anaconda_file/bin:$PATH
  6. 開啟終端,輸入python,如果看到有Anaconda資訊就證明安裝完畢
  7. 如果你還有什麼包沒有安裝上,可以執行conda install *** 來進行安裝(***代表包名稱),如果某個包版本不是最新的,執行conda update *** 就可以了。

PyTorch

接下來在Anaconda下安裝PyTorch,最好是自己新建立一個虛擬環境,預設是在base下安裝。

  1. 建立新的虛擬環境:conda create -n your_environment_name python=python3.6(這個自己選擇,也就是Anoconda的好處)
  2. 進入虛擬環境搭建pytorch環境:source activate your_environment_name
  3. 看到終端顯示(your_environment_name)/$ 就表明成功進入
  4. 安裝:conda install pytorch torchvision
  5. torchvision是pytorch一個很有用的包提供主流的model,和常用資料集;對 torch.utils.data.Dataset進行擴充,針對不同類別影象放入不同資料夾的資料進行讀取, torchvision.datasets.ImageFolder是torch.utils.data.Dataset 的子類;提供transforms。

References

https://blog.csdn.net/u013069552/article/details/78349716
https://www.jianshu.com/p/c78cbbf2a6d8
https://blog.csdn.net/u014380165/article/details/79119664