3 北京超算云计算平台深度学习环境配置笔记

06-17 1068阅读

点击链接https://cloud.blsc.cn/进入网页版或下载客户端,使用北京超级云计算中心账号登陆

环境信息加载总结
#!/bin/bash
#加载模块
module load anaconda/2021.05
module load cuda/11.1
module load gcc/7.3
#激活环境
source activate opennmmlab_mmclassification

1.SSH到服务器

点击页面上的SSH进行远程连接,点击连接后进行命令行界面

进入run目录下,run目录下有300G的内存,可以把数据和代码等各种资料存到里面

cd run

3 北京超算云计算平台深度学习环境配置笔记

2. 创建环境

首先在命令行输入,查看预先安装的工具

module avail

3 北京超算云计算平台深度学习环境配置笔记

安装anaconda 2021.05版本

module load anaconda/2021.05

创建一个python环境

conda create --name mmclassification python=3.8

3 北京超算云计算平台深度学习环境配置笔记

激活环境

source activate mmclassification

安装pytorch工具,超算3090显卡支持的最低版本为cuda11.1,要安装cuda11.1以上版本

pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

3 北京超算云计算平台深度学习环境配置笔记

查看预先安装的工具

module avail

加载cuda/11.1

module load cuda/11.1

3 北京超算云计算平台深度学习环境配置笔记

安装MMCV

pip install mmcv-full==1.7.0 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.10/index.html

3 北京超算云计算平台深度学习环境配置笔记

安装mmclassification

新建文件夹openmmlab和mmclassification

mkdir openmmlab
mkdir mmclassification

3 北京超算云计算平台深度学习环境配置笔记

进入openmmlab文件夹下载mmclassification

cd openmmlab
git clone https://github.com/open-mmlab/mmclassification.git

3 北京超算云计算平台深度学习环境配置笔记

进入~/run/mmclassification/文件夹下

cd ..
cd mmclassification/

复制~/run/openmmlab/下文件夹mmclassification到当前文件夹下

cp -r ~/run/openmmlab/mmclassification/ ./

3 北京超算云计算平台深度学习环境配置笔记

3 北京超算云计算平台深度学习环境配置笔记然后进入目录里面来,即 ~/run/mmclassification/ mmclassification/目录下

cd mmclassification/

3 北京超算云计算平台深度学习环境配置笔记

查看gcc版本gcc --version

加载gcc/7.3

module load gcc/7.3

3 北京超算云计算平台深度学习环境配置笔记

进行编译安装,需要的依赖包也会自动安装

pip install -e .

查看软件包

pip list

3 北京超算云计算平台深度学习环境配置笔记

至此安装完成

3. 整理 flower 数据集

新建一个data文件夹,即 ~/run/mmclassification/ mmclassification/data

mkdir data

下载数据集使用快传拖拽放到 ~/run/mmclassification/ mmclassification/data目录下

进入data并解压

cd data
unzip flower_dataset.zip

a.将数据集按 8:2 的比例划分成训练和验证子数据集,并将数据集整理成 ImageNet 的格式,通过 Python 或其他脚本程序完成

复制脚本代码到当前目录下,即data目录

cp ~/run/openmmlab/data_split.py ./

3 北京超算云计算平台深度学习环境配置笔记

划分数据集时,使用绝对路径,输入pwd查看

python data_split.py /HOME/scz0be5/run/mmclassification/mmclassification/data/flower_dataset /HOME/scz0be5/run/mmclassification/mmclassification/data/flower

3 北京超算云计算平台深度学习环境配置笔记

查看flower目录下的文件

ls flower

3 北京超算云计算平台深度学习环境配置笔记

回上一级目录,新建configs/resnet18文件夹,复制配置文件~/run/openmmlab/resnet18_b32_flower.py到configs/resnet18/文件夹下

cd ..
mkdir configs/resnet18
cp ~/run/openmmlab/resnet18_b32_flower.py ./configs/resnet18/
https://aicarrier.feishu.cn/docx/QMRzd0NoxokuKvxNfS3car1EnHh
VPS购买请点击我

文章版权声明:除非注明,否则均为主机测评原创文章,转载或复制请以超链接形式并注明出处。

目录[+]