服务器中anaconda配置torch的gpu环境

2023年11月13日 15:01 ry 299

由于最近要跑深度学习的代码,而这个代码对显存和gpu性能的要求比较高,刚好实验室有3090的显卡,要了账号开始配置环境,我这个代码的要求的环境是python3.7.7  torch1.31.1+cu116 torchvision==0.14.1+cu116,拿了账号登录发现默认的python环境是3.8,话不多说,上anaconda,直接到官网下载最新的anacodna,我服务器是linux系统的,直接下载anaconda.sh文件,上传到服务器,切换到anaconda.sh文件,使用命令

bash Anaconda3-2023.09-0-Linux-x86_64.sh

一直按enter/回车,注意慢点按,其中会出现选项,遇到输入yes就行了。然后使用命令

conda info --envs

查看虚拟环境判断是否安装成功,如果看到base就表示成功了,接下来我们使用命令创建python3.7.7的环境,命名为mytorch

conda create -n mytorch python=3.7.7

装完后再次使用命令查看,如图所示

(mytorch) student16@c508:~/lyp_torch$ conda info --envs
# conda environments:
#
base                     /home/student/student16/anaconda3
mytorch               *  /home/student/student16/anaconda3/envs/mytorch

然后切换到我们刚创建的mytorch的虚拟环境,命令如下

source activate mytorch

激活了创建的虚拟环境,我们看python3.7.7环境是否安装了,

(mytorch) student16@c508:~/lyp_torch$ python3
Python 3.7.7 (default, May  7 2020, 21:25:33) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> 

安装成功了,接下来我们到https://download.pytorch.org/whl  这个网址去下载torch和torchvision版本,然后上传到服务器,接下来使用pip3下载即可,如图所示

我们用代码测试下,如下所示

(mytorch) student16@c508:~/lyp_torch$ python3
Python 3.7.7 (default, May  7 2020, 21:25:33) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True

完美解决!

如果上述代码帮助您很多,可以打赏下以减少服务器的开支吗,万分感谢!

欢迎发表评论~

点击此处登录后即可评论


评论列表
2023年11月16日 13:25 ry: 回复
本人承接各种软件定制,数据采集,接口搭建,网站开发,环境配置的杂活,需要的可以联系我 qq:1449917271 微信:liuyoudyping


赣ICP备2021001574号-1

赣公网安备 36092402000079号