mirror of https://github.com/open-mmlab/mmcv.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
944 B
23 lines
944 B
ARG PYTORCH="1.8.1" |
|
ARG CUDA="10.2" |
|
ARG CUDNN="7" |
|
|
|
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel |
|
|
|
# To fix GPG key error when running apt-get update |
|
RUN rm /etc/apt/sources.list.d/cuda.list \ |
|
&& rm /etc/apt/sources.list.d/nvidia-ml.list \ |
|
&& apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub \ |
|
&& apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub |
|
|
|
# Install system dependencies for opencv-python |
|
RUN apt-get update && apt-get install -y libgl1 libglib2.0-0 \ |
|
&& apt-get clean \ |
|
&& rm -rf /var/lib/apt/lists/* |
|
|
|
# Install mmcv-full |
|
ARG MMCV="" |
|
RUN if [ "${MMCV}" = "" ]; then pip install -U openmim && mim install mmcv-full; else pip install -U openmim && mim install mmcv-full==${MMCV}; fi |
|
|
|
# Verify the installation |
|
RUN python -c 'import mmcv;print(mmcv.__version__)'
|
|
|