服务器配置pytorch环境

服务器Ubuntu 16.04 LTS版本

 

由于是校外连接,因此需挂上校园VPN,然后我使用的是MobaXteram软件,用来连接服务器。关于这个软件的使用指南和下载可以参考:https://zhuanlan.zhihu.com/p/61013117,该软件免费版已经提供了很多功能,但好像文本编辑设置需要开会员才能保存。

连接上服务器后,进入正式的环境配置:

1、安装Anaconda

可以用 wget 命令下载:

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.1.0-Linux-x86_64.sh

也可以直接用:

wget https://repo.continuum.io/archive/Anaconda3-5.0.1-Linux-x86_64.sh

下载好后,安装就行,

bash Anaconda3-5.0.1-Linux-x86_64.sh

然后就一路yes,最后可能会问你是否是否下载VS Code,这个选no就安装成功了(也有可能不会遇到这个选项)。

重启服务器后进行验证。

验证Anaconda是否安装成功,在命令行输入python,若显示python版本后带了Anaconda标识就证明Anaconda安装成功。

服务器配置pytorch环境_第1张图片

 

2、第二步是安装pytorch

直接去pytorch官网pytorch.org寻找对应的版本,可以提前查看一下你的服务器的cuda版本,

cat  /usr/local/cuda/version.txt

服务器配置pytorch环境_第2张图片

可能有的人pip太慢,有的人conda不行,几个Command都试试,有一个可以就行。这里我选择的是CUDA版本为10.1。(起初选择10.2的版本,然后显示torch.cuda.is_available()返回false,可能是因为我的服务器cat命令显示的cuda版本为10.0,而nvidia -sim命令显示的cuda版本是10.2,后来我使用rm -r *命令格式化后选择安装10.1版本的就没问题了,不明白)

 

这里如果都不行,网上建议说要换源(未尝试成功)。

安装好pytorch后,要重启一下服务器。

然后验证pytorch是否安装成功,输入python后:

服务器配置pytorch环境_第3张图片

"__version__"前后都是双下划线。

 

到此为止,pytorch配置成功,在python下输入print(torch.cuda.is_available()),若返回true,则可使用gpu加速,返回false,emmmm,慢慢百度吧。。。实在不行,就rm -r *,重新来过,可能啥问题都解决了(我就是这样做的,嘤嘤嘤)。

你可能感兴趣的:(随笔,ubuntu,anaconda)