FROM nvcr.io/nvidia/pytorch:23.05-py3
RUN apt-get update
RUN apt-get install -y tmux htop

RUN git clone https://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@github.com/woxihuanjiangguo/navsim_ours.git /navsim_ours
WORKDIR /navsim_ours

ENV HYDRA_FULL_ERROR=1
ENV NUPLAN_MAP_VERSION="nuplan-maps-v1.0"
ENV NUPLAN_MAPS_ROOT="/zhenxinl_nuplan/navsim_workspace/dataset/maps"
ENV NAVSIM_EXP_ROOT="/zhenxinl_nuplan/navsim_workspace/exp"
ENV NAVSIM_DEVKIT_ROOT="/navsim_ours"
ENV NAVSIM_TRAJPDM_ROOT="/zhenxinl_nuplan/navsim_workspace/dataset/traj_pdm"
ENV OPENSCENE_DATA_ROOT="/zhenxinl_nuplan/navsim_workspace/dataset"
ENV CUDA_TOOLKIT_ROOT_DIR=$CUDA_HOME
ENV CFLAGS="-I$CUDA_HOME/include $CFLAGS"

RUN pip uninstall torch torchvision torchaudio -y
RUN pip3 install torch torchvision torchaudio
RUN pip install openmim
RUN mim install mmdet==2.28.2
RUN pip install spconv-cu120
RUN pip install numba
RUN pip install -e .