MMClassification 的 docker 镜像，基于官方文档的步骤完成构建，容器内部的 ubuntu 已更换成华为镜像源。
64
MMClassification 是一款基于 PyTorch 的开源图像分类工具箱, 集成了常用的图像分类网络，将数据加载，模型骨架，训练调参，流程等封装为模块调用，便于在模型间进行转换和比较，也高效简洁的实现了参数调整。MMClassification 是 OpenMMLab 项目的成员之一。
docker pull seeleclover/mmclassification
docker run -itd \
--gpus=all \
--name {CONTAINER NAME} \
-v {DATA_DIR}:/data:rw \
seeleclover/mmclassification:{TAG NAME} \
bash
docker exec -it {CONTAINER NAME} bash
Content type
Image
Digest
Size
4.6 GB
Last updated
about 4 years ago
docker pull seeleclover/mmclassification:v1.0_pytorch1.8.1_cuda10.2_cudnn7