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

# Set non-interactive mode to avoid prompts
ARG DEBIAN_FRONTEND=noninteractive

# setup timezone
ENV TZ=US/Pacific
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone

SHELL ["/bin/bash", "-c"]

# Install requried libraries
RUN apt-get update && apt-get install -y --no-install-recommends \
    wget \
    git \
    vim \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender1 \
    libgl1-mesa-glx


# Install Miniconda
RUN wget \
    https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh \
    && mkdir /root/.conda \
    && bash Miniconda3-latest-Linux-x86_64.sh -b \
    && rm -f Miniconda3-latest-Linux-x86_64.sh 

ENV CONDA_DEFAULT_ENV=${project}
ENV CONDA_PREFIX=/root/miniconda3/envs/$CONDA_DEFAULT_ENV
ENV PATH=/root/miniconda3/bin:$CONDA_PREFIX/bin:$PATH

# install python 3.8
RUN conda install python=3.8
RUN alias python='/root/miniconda3/envs/bin/python3.8'

# Set environment and working directory
ENV CUDA_HOME=/usr/local/cuda
ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$CUDA_HOME/extras/CUPTI/lib64/:$LD_LIBRARY_PATH
ENV PATH=$CUDA_HOME/bin:$PATH
ENV CFLAGS="-I$CUDA_HOME/include $CFLAGS"
ENV FORCE_CUDA="1"
ENV PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/root/miniconda3/envs/bin:$PATH
ENV TORCH_CUDA_ARCH_LIST="7.0+PTX"

# install pytorch
RUN pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113  -f https://download.pytorch.org/whl/torch_stable.html

# install other dependencies
RUN pip install mmcv-full==1.6.0
RUN pip install mmsegmentation==0.24.0
RUN pip install mmdet==2.24.0
# Install FB-BEV
RUN git clone https://github.com/NVlabs/FB-BEV.git && \
    cd FB-BEV && \
    pip install -e .

# install correct packages
RUN pip install numpy==1.23.5 \
    spconv \
    IPython

# Set the default command to bash
CMD ["/bin/bash"]