【記錄貼】Ubuntu16.04安裝pytorch+tensorboardX
阿新 • • 發佈:2019-01-09
為了防止自己以後需要重灌忘記怎麼安裝,以後會簡單記錄下科研過程中安裝的軟體,可能還會記錄復現程式碼過程。主要是給自己看的,不會花太多時間排版,大家不要介意哈!
在網上找的教程都好麻煩,明明直接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,完成啦!是不是很簡單!就不知道為啥大部分教程搞得那麼麻煩,感覺還都沒咋說的太清楚!