Dockerfile

FROM nvidia/cuda:11.3.1-devel-ubuntu20.04

# # updating the CUDA Linux GPG Repository Key
COPY cuda-keyring_1.0-1_all.deb .
# RUN rm /etc/apt/sources.list.d/cuda.list && rm /etc/apt/sources.list.d/nvidia-ml.list && dpkg -i cuda-keyring_1.0-1_all.deb
RUN rm /etc/apt/sources.list.d/cuda.list && dpkg -i cuda-keyring_1.0-1_all.deb

# Install basic packages
ENV TZ=Europe/London
ENV DEBIAN_FRONTEND=noninteractive
RUN DEBIAN_FRONTEND=noninteractive apt-get update && \
    apt-get install -y libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender-dev \
    wget \
    git \
    unzip \
    curl \
    nano \
    libopenexr-dev \
    openexr \
    ffmpeg \
    python3-pip \
    python3-setuptools \
    libopenblas-dev \
    && apt-get -y clean all \
    && rm -rf /var/lib/apt/lists/*

# # Install python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir --upgrade pip \
    && pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir torch==1.10.0+cu113 torchvision==0.11.0+cu113 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

# ## Install MinkowskiEngine
ENV MAX_JOBS=4
# ENV TORCH_CUDA_ARCH_LIST="6.1"
# RUN /bin/bash -c "git clone https://github.com/NVIDIA/MinkowskiEngine.git; cd MinkowskiEngine; python3 setup.py install --blas=openblas --force_cuda"
RUN pip install -U git+https://github.com/NVIDIA/MinkowskiEngine -v --no-deps --install-option="--force_cuda" --install-option="--blas=openblas" 

# SemanticKITTI API
WORKDIR /usr/src/app
RUN /bin/bash -c "curl -sL https://deb.nodesource.com/setup_18.x | bash"
RUN DEBIAN_FRONTEND=noninteractive apt-get update && \
    apt-get install -y python3-pyqt5.qtopengl \
    nodejs \
    && apt-get -y clean all \
    && rm -rf /var/lib/apt/lists/*
RUN /bin/bash -c "git clone https://github.com/PRBonn/semantic-kitti-api.git; cd semantic-kitti-api; pip install -r requirements.txt"

### K3D
RUN pip3 uninstall -y ipywidgets
RUN pip3 install ipywidgets==7.6.0
RUN pip3 install k3d
RUN /bin/bash -c "jupyter nbextension install --py --user k3d; jupyter nbextension enable --py --user k3d"

# spconv (for Voxel data generation)
RUN pip3 install spconv-cu113
# 2DPASS extra dependencies
RUN pip3 install pytorch_lightning==1.8 torch-scatter easydict
WORKDIR /usr/src/app/semantic-scene-completion