【记录贴】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，完成啦！是不是很简单！就不知道为啥大部分教程搞得那么麻烦，感觉还都没咋说的太清楚！