| FROM nvcr.io/nvidia/pytorch:23.05-py3 | |
| RUN apt-get update | |
| RUN apt-get install -y tmux htop | |
| RUN git clone https://[email protected]/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 . |