【记录】使用远程SSH配置d2l环境（含装pytorch，同时适用于本地anaconda）

news2024/12/25 9:31:50

文章目录

前言
一、从创建新环境开始
二、使用步骤
- 1.安装pytorch
- 2.安装 d2l 包
- 3.安装其他包
- 4.使用jupyter notebook

前言

记录一下如何利用使用命令行进行anaconda配置 d2l环境、pytorch并进行训练深度学习模型。

一、从创建新环境开始

如果是本地直接装一个 anaconda 软件就行，如果是像我样使用 SSH 远程连接服务器进行操作，建议服务器上安装一个 miniconda。conda中创建一个新环境的命令是：

conda  create  --name  env_name（你的环境名字） python=3.8 # 创建指定python版本

在这里插入图片描述
无脑y就是了

使用conda info --envs 可以查看环境，可以看到我们已经创建好一个名叫 d2l 的新环境了

在这里插入图片描述
使用conda actiavte d2l进入这个新环境中

二、使用步骤

1.安装pytorch

首先我们要查看一下我们显卡所支持的CUDA版本，在命令行输入nvidia-smi，然后图中画框的地方就可以看到我们显卡的CUDA支持版本，我这里这块 RTX 4090 最高支持 12.2 的 CUDA。
在这里插入图片描述

首先我们需要在torch官网中查看一下对应的torch版本。我选了中间的 CUDA 12.1 进行安装，将pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu121这行复制下来。
在这里插入图片描述
直接粘贴到命令行运行就行，然后就开始框框下载了。

安装完之后就会显示 successfully 巴拉巴拉：
在这里插入图片描述
以防万一，我们输入pip list检查一下是不是真的装上了：

2.安装 d2l 包

我这里是使用pip install d2l==0.17.6 -i https://pypi.mirrors.ustc.edu.cn/simple安装 0.17.6 版本的 d2l 包。
在这里插入图片描述
同理使用pip list检查一下，可以看到是安装成功的

同时推荐用pip install jupyter -i https://pypi.mirrors.ustc.edu.cn/simple安装一下jupyter notebook，毕竟mu神视频一直用得是jupyter，也挺好使。