2025-11-14 17:06:09 +08:00
|
|
|
|
FROM nvcr.io/nvidia/cuda:11.3.1-devel-ubuntu20.04
|
2022-09-27 06:51:16 +08:00
|
|
|
|
|
|
|
|
|
|
RUN apt-get update && apt-get install wget -yq
|
|
|
|
|
|
RUN apt-get install build-essential g++ gcc -y
|
|
|
|
|
|
ENV DEBIAN_FRONTEND noninteractive
|
|
|
|
|
|
RUN apt-get install libgl1-mesa-glx libglib2.0-0 -y
|
|
|
|
|
|
RUN apt-get install openmpi-bin openmpi-common libopenmpi-dev libgtk2.0-dev git -y
|
|
|
|
|
|
|
|
|
|
|
|
# Install miniconda
|
|
|
|
|
|
ENV CONDA_DIR /opt/conda
|
2025-11-14 17:06:09 +08:00
|
|
|
|
RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-py38_4.12.0-Linux-x86_64.sh -O ~/miniconda.sh && \
|
2022-09-27 06:51:16 +08:00
|
|
|
|
/bin/bash ~/miniconda.sh -b -p /opt/conda
|
2025-11-14 17:06:09 +08:00
|
|
|
|
|
|
|
|
|
|
|
2022-09-27 06:51:16 +08:00
|
|
|
|
# Put conda in path so we can use conda activate
|
|
|
|
|
|
ENV PATH=$CONDA_DIR/bin:$PATH
|
2025-11-14 17:06:09 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#RUN conda config --add channels conda-forge
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
RUN conda config --add channels pytorch
|
|
|
|
|
|
RUN conda config --add channels nvidia
|
|
|
|
|
|
RUN conda config --set remote_connect_timeout_secs 60.0
|
|
|
|
|
|
RUN conda config --set remote_read_timeout_secs 120.0
|
|
|
|
|
|
RUN conda config --set remote_max_retries 5
|
|
|
|
|
|
|
|
|
|
|
|
RUN conda config --set show_channel_urls yes
|
|
|
|
|
|
|
|
|
|
|
|
RUN conda config --set channel_priority flexible
|
|
|
|
|
|
# <20><><EFBFBD><EFBFBD>ֱ<EFBFBD>Ӱ<EFBFBD>װpython<6F><6E><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʹ<EFBFBD><CAB9>conda-forge<67>İ<EFBFBD>
|
|
|
|
|
|
RUN conda install python=3.8 -y
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#RUN conda install python=3.8 -y
|
|
|
|
|
|
RUN conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3
|
2022-09-27 06:51:16 +08:00
|
|
|
|
RUN pip install Pillow==8.4.0
|
|
|
|
|
|
RUN pip install tqdm
|
|
|
|
|
|
RUN pip install torchpack
|
|
|
|
|
|
RUN pip install mmcv==1.4.0 mmcv-full==1.4.0 mmdet==2.20.0
|
|
|
|
|
|
RUN pip install nuscenes-devkit
|
|
|
|
|
|
RUN pip install mpi4py==3.0.3
|
|
|
|
|
|
RUN pip install numba==0.48.0
|