linux服务器配置pytorch环境

安装anaconda

从官网下载anaconda

打开ananconda官网链接ananconda官网
linux服务器配置pytorch环境
点击最下面小企鹅的图标(这个是linux系统的)然后点击下载。
linux服务器配置pytorch环境

点击第一个链接下载。下载后,上传到自己的服务器。
linux服务器配置pytorch环境

安装anaconda

首先我们先进入刚才放anaconda安装包的位置。
linux服务器配置pytorch环境
输入以下命令

chmod +x Anaconda3-xxxxx-Linux-x86_64.sh
./Anaconda3-xxxxx-Linux-x86_64.sh	# or bash Anaconda3-xxxxx-Linux-x86_64.sh

安装路径默认是home,不过最好装在data路径下,这时候需要我们指定安装路径。
linux服务器配置pytorch环境
可能需要在路径后面加个-u。
然后等待安装即可。安装完成之后重启服务器即可看到base环境
linux服务器配置pytorch环境

安装pytorch

创造新环境

anacoda给了一个默认的base环境,但是我们一般不再base环境进行配置。我们再创一个其他环境来使用。

conda create --name yourEnv python=3.8

这个python的版本需要和cuda版本对应起来。查看cuda版本的命令如下

nvidia-smi

linux服务器配置pytorch环境
cuda,python,pytorch对应的版本表格如下:
linux服务器配置pytorch环境
然后就可以去pytorch官网下载对应的torch版本了
linux服务器配置pytorch环境
如果cuda版本上面没有可以点这里下载以前的版本。
找到对应的版本后,复制命令进行相应的下载。
安装之后进入python 输入import torch不报错说明成功

linux服务器配置pytorch环境
为了验证GPU是否可用,可以接着输入如下指令:

torch.cuda.is_available ()

linux服务器配置pytorch环境
显示True即表示之前安装的python,pytorch,cuda版本都是匹配的。如果是False需要检查一下上面的版本对应问题。

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(1)
xiaoxingxing的头像xiaoxingxing管理团队
上一篇 2022年4月2日
下一篇 2022年4月4日

相关推荐