1. 程式人生 > >【記錄貼】Ubuntu16.04安裝pytorch+tensorboardX

【記錄貼】Ubuntu16.04安裝pytorch+tensorboardX

為了防止自己以後需要重灌忘記怎麼安裝,以後會簡單記錄下科研過程中安裝的軟體,可能還會記錄復現程式碼過程。主要是給自己看的,不會花太多時間排版,大家不要介意哈!

在網上找的教程都好麻煩,明明直接pip安裝簡單又好用,自己安裝完全能用,所以記錄一下。

師兄說安裝pytorch要先安裝torchvision,因為torchvision安裝的時候會自動安裝最新版本的torch,如果自己先安裝了低版本的,就會被覆蓋掉。

1.安裝torchvision

系統上裝了兩個版本的python,2.7和3.5,這次安裝在python3.5下面。所以要用pip3,如果是python2下面安裝,就直接用pip就好了!

sudo pip3 install tochvision

這裡會自動安裝一個torch,版本好像是1.0.0,解除安裝的時候帶上版本號。

2.解除安裝自動安裝的torch

sudo pip3 uninstall torch==1.0.0

3.安裝需要的torch版本

sudo pip3 install torch==0.4.1

4.安裝視覺化工具tensorboardX

sudo pip3 install tensorboardX

5.安裝成功檢視版本

python3
import torch
print(torch.__version__)  #注意是雙下劃線

如果是python2安裝,就是

python
import torch
print(torch.__version__)  #注意是雙下劃線

OK,完成啦!是不是很簡單!就不知道為啥大部分教程搞得那麼麻煩,感覺還都沒咋說的太清楚!