1. 运行docker
基本命令:
sudo docker ps -a (列出所有容器状态)
sudo docker run -dit -v /hdd/lyh/mask2former:/mask --gpus '"device=0,1"' --shm-size 16G --name mask 11.1:v6 (创建docker容器,用镜像11.1:v6,容器名称为mask,将主机目录 /hdd/lyh/mask2former
挂载到容器内的 /mask
目录。这允许容器访问和使用主机上的该目录中的文件和数据。)
返回的是容器id
sudo docker images(列出本地所有docker镜像)
sudo docker exec -it mask bash(运行容器)
sudo docker stop mask(停止容器)
2. conda虚拟环境
在容器中,执行这两条指令
conda activate mask
cd /mask/Mask2Former
3. 训练和测试
训练命令:
python train_net.py --num-gpus 2 --config-file ./configs/s4_maskformer2_R50_bs16_90k.yaml
测试命令:
python train_net.py --num-gpus 2 --config-file ./configs/s4_maskformer2_R50_bs16_90k.yaml --eval-only MODEL.WEIGHTS path/to/your/.pth(在训练命令的基础上,多了eval-only和.pth文件的路径)