反向commit 镜像:
docker commit 74d9893ccb29 kevinchina/deeplearning:fasterrcnn_train_v1
docker push kevinchina/deeplearning:fasterrcnn_train_v1
重启容器:
docker run --gpus all -it -v $PWD:/wkp --shm-size=64g kevinchina/deeplearning:fasterrcnn_train_v1 bash
训练:
conda activate py38
cd /deep-learning-for-image-processing/pytorch_object_detection/faster_rcnn/
python train_mobilenetv2.py
启动成功:
一轮训练完成后的验证: