Question about docker container on {sh ./make.sh}

fundamentalvision / Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.

Apache License 2.0

3.15k stars 513 forks source link

Question about docker container on {sh ./make.sh} #135

Open Ancrilin opened 2 years ago

Ancrilin commented 2 years ago

Hi, I want to compile on a pytorch container, but the cuda directory don't exist in my container (/usr/local/cuda or /opt/cuda). I want to know how to run sh ./make.sh. I pull the container from dockerhub: docker pull pytorch/pytorch:1.11.0-cuda11.3-cudnn8-devel.

ayushjain1144 commented 2 years ago

Hi, A solution for me was to add a line like the following in the dockerfile:

CMD sh Baseline_Controller_Eval_docker.sh

And in the Baseline_Controller_Eval_docker.sh, I put:

set -e

cd Deformable-DETR/models/ops &&  rm -rf build && sh make.sh && cd ../../../..