facebookresearch/maskrcnn-benchmark---Docker Runtime Error: Not Compiled with GPU support

方法解决来源是:
https://github.com/facebookresearch/maskrcnn-benchmark/issues/167
facebookresearch/maskrcnn-benchmark---Docker Runtime Error: Not Compiled with GPU support_第1张图片
我已经把这个东西镜像做好了,并且源码和镜像都已经上传。

  • 新版docker镜像是:https://hub.docker.com/r/xxxxxxxxxxxxxxxxxxxx/maskrcnn/tags
    使用方法是:sudo docker run --shm-size 8G --runtime=nvidia -it -p 8888:8888 -e PASSWORD='xxxx' -v /home/jerry/workspace/:/workspace -v /home/jerry/workspace/maskrcnn-benchmark:/maskrcnn-benchmark xxxxxxxxxxxxxxxxxxxx/maskrcnn:env0.1
  • 整理好的源代码是:
    https://github.com/GuoLiuFang/maskrcnn-benchmark-lfs
  • 根据自己的实际情况更改数据路径:
ln -sf /workspace/datasets/datasets/coco/annotations datasets/coco/annotations
ln -sf /workspace/datasets/datasets/coco/val2014 datasets/coco/val2014
  • 启动jupyter,运行测试demo
source activate py36
jupyter notebook --allow-root -y --no-browser --ip=0.0.0.0 --config=/maskrcnn-benchmark/docker/docker-jupyter/jupyter_notebook_config.py

  • 进行测试训练
python tools/train_net.py --config-file "configs/e2e_mask_rcnn_R_50_FPN_1x.yaml"
python tools/train_net.py --config-file "configs/e2e_mask_rcnn_R_101_FPN_1x.yaml"


欢迎交流联系:QQ:909104374

你可能感兴趣的:(OCR,贝壳,R芳,NLP)