search

facebookresearch / Detectron

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.
Apache License 2.0
26.24k stars 5.45k forks source link

How can I use detectron with pytorch? #757

Open BasselAli1 opened 5 years ago

BasselAli1 commented 5 years ago

Hi guys :sweat_smile: :sweat_smile: I have a question but I don't know if this the right place to ask or not. I am using pytorch and I wanted to use detectron and they are saying that I have to install caffe2 first and I did so but when type from caffe2.python import core, I see this WARNING:root:This caffe2 python run does not have GPU support. Will run in CPU only mode. CRITICAL:root:Cannot load caffe2.python. Error: /lib64/libc.so.6: version 'GLIBC_2.14' not found (required by /home/qwe123/data/conda_packages/lib/python3.6/site-packages/caffe2/python/caffe2_pybind11_state.cpython-36m-x86_64-linux-gnu.so) I tried other solutions I found on here but I could not get this to work. I only installed caffe2 because I wanted to use detectron, Can I use it with pytorch without the need of caffe2? My version of pytorch of 4.0, Should I download pytorch version 1.0 to use detectron with it?

guohaoyu110 commented 5 years ago

you can just play with another version of Mask RCNN https://github.com/facebookresearch/maskrcnn-benchmark

planetceres commented 5 years ago

Doesn't INSTALL.md#caffe2 link to the pytorch installation here? If this isn't correct, it seems like an appropriate url for Caffe2 installation should be referenced instead.