File tree Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change 11ARG PYTORCH="1.11.0"
22ARG CUDA="11.3"
33ARG CUDNN="8"
4- ARG MMCV="1.4.8"
54
65FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
76
87ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0+PTX"
98ENV TORCH_NVCC_FLAGS="-Xfatbin -compress-all"
109ENV CMAKE_PREFIX_PATH="$(dirname $(which conda))/../"
1110
11+ # To fix GPG key error when running apt-get update
12+ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
13+ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
14+
1215RUN apt-get update && apt-get install -y git ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 \
13- && apt-get clean \
14- && rm -rf /var/lib/apt/lists/*
16+ && apt-get clean \
17+ && rm -rf /var/lib/apt/lists/*
1518
1619RUN conda clean --all
1720
1821# Install MMCV
1922ARG PYTORCH
2023ARG CUDA
2124ARG MMCV
22- RUN ["/bin/bash" , "-c" , "pip install --no-cache-dir mmcv-full==${MMCV} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html" ]
25+ RUN ["/bin/bash" , "-c" , "pip install --no-cache-dir mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html" ]
2326
2427# Install MMSegmentation
2528RUN git clone https://github.com/open-mmlab/mmsegmentation.git /mmsegmentation
You can’t perform that action at this time.
0 commit comments