Open mzahran001 opened 6 years ago
Hi @moh3th1, I don't quite follow the steps you're taking in order to use the Detectron docker image.
As described in INSTALL.md, to build the Detectron docker image it is sufficient to:
cd $DETECTRON/docker
docker build -t detectron:c2-cuda9-cudnn7 .
You can then run the demo using:
nvidia-docker run --rm -it detectron:c2-cuda9-cudnn7 python2 tools/infer_simple.py \
--cfg configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml \
--output-dir /tmp/detectron-visualizations \
--image-ext jpg \
--wts https://s3-us-west-2.amazonaws.com/detectron/35861858/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml.02_32_51.SgT4y1cO/output/train/coco_2014_train:coco_2014_valminusminival/generalized_rcnn/model_final.pkl \
demo
Hi @ir413 . Thanks for your response. Unfortunately it produce the same error. Is there is a way to force caffe2 use CPU instead of GPU?
Unfortunately it produce the same error.
Just to be sure that an older cached version is not being used, please try rebuilding the Detectron image with the --no-cache
option:
cd $DETECTRON/docker
docker build --no-cache -t detectron:c2-cuda9-cudnn7 .
If the error is still there, please paste the complete output from both building and running the image.
Is there is a way to force caffe2 use CPU instead of GPU?
Detectron currently requires Caffe2 with GPU support.
same error, not using docker image.
Same error here, does someone have a solution for this?
Expected results
Actual results
What did you observe instead?
Detailed steps to reproduce
I am using caffe2 + Detecron docker image but when I am trying to Inference with Pretrained Models I got these errors. Caffe2 latest Docker image using GPU support:
Build the image:
make new docker
Install the COCO API
build the python modules
Making Inferences
System information