linux服务器配置pytorch环境

本文介绍了如何在Linux服务器上下载并安装Anaconda,包括更改安装路径,创建新环境。接着详细阐述了如何根据CUDA版本选择合适的Python和PyTorch版本,并提供了验证安装成功的步骤,确保Python、PyTorch与CUDA的兼容性。
摘要由CSDN通过智能技术生成

安装anaconda

从官网下载anaconda

打开ananconda官网链接ananconda官网
在这里插入图片描述
点击最下面小企鹅的图标(这个是linux系统的)然后点击下载。
在这里插入图片描述

点击第一个链接即可下载。下载完之后上传到自己服务器。
在这里插入图片描述

安装anaconda

首先我们先进入刚才放anaconda安装包的位置。
在这里插入图片描述
输入如下命令

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

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

安装pytorch

创建新的环境

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

conda create --name yourEnv python=3.8

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

nvidia-smi

在这里插入图片描述
cuda,python,pytorch对应的版本表格如下:
在这里插入图片描述
然后就可以去pytorch官网下载对应的torch版本了
在这里插入图片描述
如果cuda版本上面没有可以点这里下载以前的版本。
找到对应的版本之后复制命令即可相应的下载。
安装之后进入python 输入import torch不报错说明成功

在这里插入图片描述
为了验证GPU是否可用,可以接着输入如下指令:

torch.cuda.is_available ()

在这里插入图片描述
显示True即表示之前安装的python,pytorch,cuda版本都是匹配的。如果是False需要检查一下上面的版本对应问题。

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值