diff --git a/docker/cu116.Dockerfile b/docker/cu116.Dockerfile new file mode 100644 index 000000000..f68e4bdf7 --- /dev/null +++ b/docker/cu116.Dockerfile @@ -0,0 +1,83 @@ +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