search

airsplay / py-bottom-up-attention

PyTorch bottom-up attention with Detectron2
Apache License 2.0
229 stars 56 forks source link

Requires AMD ROCM to work (no hipcc found), merged from here into fb/detectron2 as solution #8

Open faizanahemad opened 4 years ago

faizanahemad commented 4 years ago

Problem

Tried Installing on a Linux GPU V100 cuda 10.1 machine with GLIBC 2.17 , This repo requires AMD ROCM (no hipcc found was the gcc error). AMD ROCM needs multiple dependencies which need GLIBC 2.22.

In the most recent detectron2 setup.py They have made ROCM optional.

Solution

I ported the changes in this repo to most recent detectron2 master (dated 21st June 2020, 0de7e8c) at here https://github.com/faizanahemad/detectron2.git . This removes ROCM need and also updates Detectron2. See demo/demo_caffe_frcnn_feature_extraction_attr.py or demo/demo_caffe_frcnn_feature_extraction_attr.ipynb for feature extraction for LXMERT.

Install as python -m pip install 'git+https://github.com/faizanahemad/detectron2.git'