OpenMMLab 实战营打卡 - 第3课

一、MMClassfication

MMClassfication 已经支持大量的模型，也支持了主流的图像分类数据集。 

MMClassfication仓库地址：https://github.com/open-mmlab/mmclassification

MMClassfication教程地址：https://mmclassification.readthedocs.io/en/latest/

MMClassfication主要通过构建配置文件进行，简化了直接手写Pytorch代码的过程。

 

 二、使用北京超算平台配置MMClassfication环境

1、官网(北京超级云计算中心)下载客户端 并打开  , 输入账号密码

2、打开SSH

3、进入run目录

cd run/

 4、加载Anaconda工具

module avail
module load anaconda/2021.05

5、创建MMCls环境

创建虚拟环境

conda create --name mmclassfication python=3.8

激活虚拟环境

source activate mmclassfication

安装pytorch

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

 加载cuda

module load cuda/11.1

安装mmcv-full

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

安装mmclassfication

mkdir mmclassfication
cd mmclassfication
git clone https://github.com/open-mmlab/mmclassification.git
cd mmclassfication
module load gcc/7.3
pip install -v -e .