mirror of https://github.com/FoundationVision/GLEE
33 lines
1.1 KiB
Docker
33 lines
1.1 KiB
Docker
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
# This file defines a container that compiles the C++ examples of detectron2.
|
|
# See docker/README.md for usage.
|
|
|
|
# Depends on the image produced by "./Dockerfile"
|
|
FROM detectron2:v0
|
|
|
|
USER appuser
|
|
ENV HOME=/home/appuser
|
|
WORKDIR $HOME
|
|
|
|
# Let torchvision find libtorch
|
|
ENV CMAKE_PREFIX_PATH=$HOME/.local/lib/python3.6/site-packages/torch/
|
|
|
|
RUN sudo apt-get update && sudo apt-get install libopencv-dev --yes
|
|
|
|
# install libtorchvision
|
|
RUN git clone --branch v0.11.1 https://github.com/pytorch/vision/
|
|
RUN mkdir vision/build && cd vision/build && \
|
|
cmake .. -DCMAKE_INSTALL_PREFIX=$HOME/.local -DCMAKE_BUILD_TYPE=Release -DWITH_CUDA=on -DTORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST && \
|
|
make -j && make install
|
|
|
|
# make our installation take effect
|
|
ENV CPATH=$HOME/.local/include \
|
|
LIBRARY_PATH=$HOME/.local/lib \
|
|
LD_LIBRARY_PATH=$HOME/.local/lib
|
|
|
|
|
|
# build C++ examples of detectron2
|
|
RUN cd detectron2_repo/tools/deploy && mkdir build && cd build && \
|
|
cmake -DTORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST .. && make
|
|
# binaries will be available under tools/deploy/build
|