1. 程式人生 > >Ubuntu下PyTorch 安裝 (基於conda)

Ubuntu下PyTorch 安裝 (基於conda)

最近準備學習一下PyTorch,裝的時候也是各種坑坑,記錄一下過程以供查閱。

1、安裝conda命令(anaconda)
Anaconda指的是一個開源的Python發行版本,其包含了Conda、Python以及一大堆安裝好的工具包,比如:numpy、pandas等,下載檔案比較大(約 617 MB)。
Miniconda是較小的發行版僅僅包括Conda、Python。
因為有很多庫沒有安裝,現在直接選擇安裝Anaconda。
1)在Anaconda官網https://www.anaconda.com/download/根據需要下載你所需要的版本。
我選擇的是linux、python2.7、64位
在這裡插入圖片描述
2)下載好後,上傳(指令rz)到linux上,執行下列命令進行安裝:
bash Anaconda2-5.3.0-Linux-x86_64.sh
安裝過程中會顯示配置路徑 Prefix=/home/search/anaconda2/,記下這個路徑;
還會詢問需不需要系統初始化,大概是你進入系統的時候要不要預設使用它?(大概這麼個意思),如果選擇yes,.bashrc檔案就會被修改。
其他按照需要y/n就好。
3)安裝完成後,還需自己設定環境變數
vi /etc/profile 開啟profile檔案
在檔案末尾新增一行 export PATH=/home/search/anaconda2/bin:$PATH 其中/home/search/anaconda2 替換成你自己的配置路徑。
在這裡插入圖片描述


但是 profile檔案是隻讀檔案,w!都無法儲存。
使用指令 :w ! sudo tee % > /dev/null ,之後選項選擇L,可以將新修改的檔案載入緩衝區。
4) 重啟 linux,安裝成功
輸入python
在這裡插入圖片描述
指令 conda --version 檢視版本號
在這裡插入圖片描述
2、安裝PyTorch
登入官網https://pytorch.org/,選擇適合自己的指令
在這裡插入圖片描述
安裝過程顯示:
在這裡插入圖片描述
安裝完成
在python 互動模式下,輸入下面指令,查詢Torch版本
import torch
print(torch.version)

插一句,安裝0.3.0指定版本的PyTorch指令(應該是對的)
conda install pytorch-cpu pytorch=0.3.0 torchvision-cpu -c pytorch
在這裡插入圖片描述