Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
OpenPCDet
Commits
255db8f0
Unverified
Commit
255db8f0
authored
Sep 15, 2023
by
Daniel Schneider
Committed by
GitHub
Sep 15, 2023
Browse files
added dockerfile for cu11.6.1 on ubuntu20.04 (#1457)
parent
bce886d6
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
83 additions
and
0 deletions
+83
-0
docker/cu116.Dockerfile
docker/cu116.Dockerfile
+83
-0
No files found.
docker/cu116.Dockerfile
0 → 100644
View file @
255db8f0
FROM
nvidia/cuda:11.6.2-devel-ubuntu20.04
# Set environment variables
ENV
NVENCODE_CFLAGS "-I/usr/local/cuda/include"
ENV
CV_VERSION=4.2.0
ENV
DEBIAN_FRONTEND=noninteractive
# Get all dependencies
RUN
apt-get update
&&
apt-get
install
-y
\
git zip unzip libssl-dev libcairo2-dev lsb-release libgoogle-glog-dev libgflags-dev libatlas-base-dev libeigen3-dev software-properties-common
\
build-essential cmake pkg-config libapr1-dev autoconf automake libtool curl libc6 libboost-all-dev debconf libomp5 libstdc++6
\
libqt5core5a libqt5xml5 libqt5gui5 libqt5widgets5 libqt5concurrent5 libqt5opengl5 libcap2 libusb-1.0-0 libatk-adaptor neovim
\
python3-pip python3-tornado python3-dev python3-numpy python3-virtualenv libpcl-dev libgoogle-glog-dev libgflags-dev libatlas-base-dev
\
libsuitesparse-dev python3-pcl pcl-tools libgtk2.0-dev libavcodec-dev libavformat-dev libswscale-dev libtbb2 libtbb-dev libjpeg-dev
\
libpng-dev libtiff-dev libdc1394-22-dev xfce4-terminal
&&
\
rm
-rf
/var/lib/apt/lists/
*
# OpenCV with CUDA support
WORKDIR
/opencv
RUN
git clone https://github.com/opencv/opencv.git
-b
$CV_VERSION
&&
\
git clone https://github.com/opencv/opencv_contrib.git
-b
$CV_VERSION
# While using OpenCV 4.2.0 we have to apply some fixes to ensure that CUDA is fully supported, thanks @https://github.com/gismo07 for this fix
RUN
mkdir
opencvfix
&&
cd
opencvfix
&&
\
git clone https://github.com/opencv/opencv.git
-b
4.5.2
&&
\
cd
opencv/cmake
&&
\
cp
-r
FindCUDA /opencv/opencv/cmake/
&&
\
cp
FindCUDA.cmake /opencv/opencv/cmake/
&&
\
cp
FindCUDNN.cmake /opencv/opencv/cmake/
&&
\
cp
OpenCVDetectCUDA.cmake /opencv/opencv/cmake/
WORKDIR
/opencv/opencv/build
RUN
cmake
-D
CMAKE_BUILD_TYPE
=
RELEASE
\
-D
CMAKE_INSTALL_PREFIX
=
/usr/local
\
-D
OPENCV_GENERATE_PKGCONFIG
=
ON
\
-D
BUILD_EXAMPLES
=
OFF
\
-D
INSTALL_PYTHON_EXAMPLES
=
OFF
\
-D
INSTALL_C_EXAMPLES
=
OFF
\
-D
PYTHON_EXECUTABLE
=
$(
which python2
)
\
-D
PYTHON3_EXECUTABLE
=
$(
which python3
)
\
-D
PYTHON3_INCLUDE_DIR
=
$(
python3
-c
"from distutils.sysconfig import get_python_inc; print(get_python_inc())"
)
\
-D
PYTHON3_PACKAGES_PATH
=
$(
python3
-c
"from distutils.sysconfig import get_python_lib; print(get_python_lib())"
)
\
-D
BUILD_opencv_python2
=
ON
\
-D
BUILD_opencv_python3
=
ON
\
-D
OPENCV_EXTRA_MODULES_PATH
=
../../opencv_contrib/modules/
\
-D
WITH_GSTREAMER
=
ON
\
-D
WITH_CUDA
=
ON
\
-D
ENABLE_PRECOMPILED_HEADERS
=
OFF
\
..
&&
\
make
-j
$(
nproc
)
&&
\
make
install
&&
\
ldconfig
&&
\
rm
-rf
/opencv
WORKDIR
/
ENV
OpenCV_DIR=/usr/share/OpenCV
# PyTorch for CUDA 11.6
RUN
pip3
install
torch
==
1.13.1+cu116
torchvision
==
0.14.1+cu116
torchaudio
==
0.13.1
--extra-index-url
https://download.pytorch.org/whl/cu116
ENV
TORCH_CUDA_ARCH_LIST="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX"
# OpenPCDet
RUN
pip3
install
numpy
==
1.23.0 llvmlite numba tensorboardX easydict pyyaml scikit-image tqdm SharedArray open3d mayavi av2 kornia pyquaternion
RUN
pip3
install
spconv-cu116
RUN
git clone https://github.com/open-mmlab/OpenPCDet.git
WORKDIR
OpenPCDet
RUN
python3 setup.py develop
WORKDIR
/
ENV
NVIDIA_VISIBLE_DEVICES="all" \
OpenCV_DIR=/usr/share/OpenCV \
NVIDIA_DRIVER_CAPABILITIES="video,compute,utility,graphics" \
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/lib:/usr/lib:/usr/local/lib \
QT_GRAPHICSSYSTEM="native"
# Build instructions: docker build -f minimal.Dockerfile -t openpcdet:cuda11 .
# Start instructions: xhost local:root && docker run -it --rm -e SDL_VIDEODRIVER=x11 -e DISPLAY=$DISPLAY --env='DISPLAY' --gpus all --ipc host --privileged --network host -p 8080:8081 -v /tmp/.X11-unix:/tmp/.X11-unix:rw -v file_locations:/storage -v /weights:/weights openpcdet:cuda11 xfce4-terminal --title=openPCDet
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment