Open dreamflasher opened 2 years ago
This command can be used to install the library after building the image: sudo docker run --rm --gpus all -it --entrypoint bash scene_graph_benchmark
But then running the code fails with:
root@1758423ba0b9:/scene_graph_benchmark# python tools/demo/demo_image.py --config_file sgg_configs/vgattr/vinvl_x152c4.yaml --img_file demo/woman_fish.jpg --save_file output/woman_fish_x152c4.obj.jpg MODEL.WEIGHT pretrained_model/vinvl_vg_x152c4.pth MODEL.ROI_HEADS.NMS_FILTER 1 MODEL.ROI_HEADS.SCORE_THRESH 0.2 TEST.IGNORE_BOX_REGRESSION False
Traceback (most recent call last):
File "tools/demo/demo_image.py", line 3, in <module>
import cv2
File "/miniconda/envs/py37/lib/python3.7/site-packages/cv2/__init__.py", line 8, in <module>
from .cv2 import *
ImportError: libGL.so.1: cannot open shared object file: No such file or directory
The Dockerfile needs: RUN apt-get install ffmpeg libsm6 libxext6 -y
But then running the code fails with:
root@1758423ba0b9:/scene_graph_benchmark# python tools/demo/demo_image.py --config_file sgg_configs/vgattr/vinvl_x152c4.yaml --img_file demo/woman_fish.jpg --save_file output/woman_fish_x152c4.obj.jpg MODEL.WEIGHT pretrained_model/vinvl_vg_x152c4.pth MODEL.ROI_HEADS.NMS_FILTER 1 MODEL.ROI_HEADS.SCORE_THRESH 0.2 TEST.IGNORE_BOX_REGRESSION False Traceback (most recent call last): File "tools/demo/demo_image.py", line 3, in <module> import cv2 File "/miniconda/envs/py37/lib/python3.7/site-packages/cv2/__init__.py", line 8, in <module> from .cv2 import * ImportError: libGL.so.1: cannot open shared object file: No such file or directory
Need to install libgl-dev
package
Yeah it needs ˋˋˋ apt-get install ffmpeg libsm6 libxext6 -y
git clone https://github.com/microsoft/scene_graph_benchmark.git && cd scene_graph_benchmark && python setup.py build develop
pip install torchvision==0.8.2+cu101 --find-links=https://download.pytorch.org/whl/torch_stable.html
mkdir pretrained_model && cd pretrained_model wget https://penzhanwu2.blob.core.windows.net/sgg/sgg_benchmark/vinvl_model_zoo/vinvl_vg_x152c4.pth cd ..
mkdir visualgenome && cd visualgenome wget https://penzhanwu2.blob.core.windows.net/sgg/sgg_benchmark/vinvl_model_zoo/VG-SGG-dicts-vgoi6-clipped.json cd .. ˋˋˋ
I'm working on a PR, but so far I wasn't able to build the library from the Docker build command, because it's missing the gpu then.
Yeah it needs ˋˋˋ apt-get install ffmpeg libsm6 libxext6 -y
git clone https://github.com/microsoft/scene_graph_benchmark.git && cd scene_graph_benchmark && python setup.py build develop
pip install torchvision==0.8.2+cu101 --find-links=https://download.pytorch.org/whl/torch_stable.html
mkdir pretrained_model && cd pretrained_model wget https://penzhanwu2.blob.core.windows.net/sgg/sgg_benchmark/vinvl_model_zoo/vinvl_vg_x152c4.pth cd ..
mkdir visualgenome && cd visualgenome wget https://penzhanwu2.blob.core.windows.net/sgg/sgg_benchmark/vinvl_model_zoo/VG-SGG-dicts-vgoi6-clipped.json cd .. ˋˋˋ
I'm working on a PR, but so far I wasn't able to build the library from the Docker build command, because it's missing the gpu then.
I'm able to build the docker image without commenting out the bottom of the docker file, but I cannot clone the repo and build it manually inside the docker container afterwards. It has " Error compiling objects for extension"
If I include the git clone and build in the docker file, then I get no CUDA driver error when building the image.
I'm wondering have you solved the issue and successfully run the docker instance? If so, could you share some details please?
This Dockerfile https://github.com/microsoft/scene_graph_benchmark/blob/main/docker/Dockerfile has the relevant lines of installing the repo commented out.
Uncommenting them and building the images results in: