由于最近要跑深度学习的代码,而这个代码对显存和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
完美解决!
如果上述代码帮助您很多,可以打赏下以减少服务器的开支吗,万分感谢!


点击此处登录后即可评论