File size: 1,682 Bytes
079c32c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime as base
WORKDIR /ding
RUN apt update \
&& apt install libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender-dev swig curl git vim gcc \g++ make wget locales dnsutils zip unzip cmake -y \
&& apt clean \
&& rm -rf /var/cache/apt/* \
&& sed -i '/en_US.UTF-8/s/^# //g' /etc/locale.gen \
&& locale-gen
ENV LANG en_US.UTF-8
ENV LANGUAGE en_US:UTF-8
ENV LC_ALL en_US.UTF-8
ADD setup.py setup.py
ADD dizoo dizoo
ADD ding ding
ADD README.md README.md
RUN python3 -m pip install --upgrade pip \
&& python3 -m pip install --ignore-installed 'PyYAML<6.0' \
&& python3 -m pip install --no-cache-dir .[fast,test]
FROM ubuntu:20.04 as doc
ENV DEBIAN_FRONTEND=noninteractive
WORKDIR /ding
RUN apt-get update && \
apt-get install --no-install-recommends -y \
python3.8 python3-pip python3.8-dev
RUN apt update \
&& apt install libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender-dev swig curl git vim gcc \g++ make wget locales dnsutils zip unzip cmake -y \
&& apt clean \
&& rm -rf /var/cache/apt/* \
&& sed -i '/en_US.UTF-8/s/^# //g' /etc/locale.gen \
&& locale-gen
ENV LANG en_US.UTF-8
ENV LANGUAGE en_US:UTF-8
ENV LC_ALL en_US.UTF-8
ADD setup.py setup.py
ADD dizoo dizoo
ADD ding ding
ADD README.md README.md
RUN python3 -m pip install --upgrade pip \
&& python3 -m pip install --ignore-installed 'PyYAML<6.0' \
&& python3 -m pip install --no-cache-dir .[fast]
WORKDIR /ding_doc
RUN git clone -b main https://github.com/opendilab/DI-engine-docs.git \
&& cd DI-engine-docs \
&& python3 -m pip install -r requirements.txt
|