【记录贴】Ubuntu16.04安装pytorch+tensorboardX

版权声明:本文属于原创,转载请注明链接! https://blog.csdn.net/echoamor/article/details/86140432

为了防止自己以后需要重装忘记怎么安装,以后会简单记录下科研过程中安装的软件,可能还会记录复现代码过程。主要是给自己看的,不会花太多时间排版,大家不要介意哈!

在网上找的教程都好麻烦,明明直接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,完成啦!是不是很简单!就不知道为啥大部分教程搞得那么麻烦,感觉还都没咋说的太清楚!

猜你喜欢

转载自blog.csdn.net/echoamor/article/details/86140432