pytorch

1
ARG UBUNTU_VERSION
2

3
FROM ubuntu:${UBUNTU_VERSION}
4

5
ARG UBUNTU_VERSION
6

7
ENV DEBIAN_FRONTEND noninteractive
8

9
# Install common dependencies (so that this step can be cached separately)
10
COPY ./common/install_base.sh install_base.sh
11
RUN bash ./install_base.sh && rm install_base.sh
12

13
# Install missing libomp-dev
14
RUN apt-get update && apt-get install -y --no-install-recommends libomp-dev && apt-get autoclean && apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
15

16
# Install user
17
COPY ./common/install_user.sh install_user.sh
18
RUN bash ./install_user.sh && rm install_user.sh
19

20
# Install conda and other packages (e.g., numpy, pytest)
21
ARG ANACONDA_PYTHON_VERSION
22
ARG CONDA_CMAKE
23
ENV ANACONDA_PYTHON_VERSION=$ANACONDA_PYTHON_VERSION
24
ENV PATH /opt/conda/envs/py_$ANACONDA_PYTHON_VERSION/bin:/opt/conda/bin:$PATH
25
COPY requirements-ci.txt /opt/conda/requirements-ci.txt
26
COPY ./common/install_conda.sh install_conda.sh
27
COPY ./common/common_utils.sh common_utils.sh
28
RUN bash ./install_conda.sh && rm install_conda.sh common_utils.sh /opt/conda/requirements-ci.txt
29

30
# Install cuda and cudnn
31
ARG CUDA_VERSION
32
COPY ./common/install_cuda.sh install_cuda.sh
33
RUN bash ./install_cuda.sh ${CUDA_VERSION} && rm install_cuda.sh
34
ENV DESIRED_CUDA ${CUDA_VERSION}
35
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:$PATH
36

37
# Note that Docker build forbids copying file outside the build context
38
COPY ./common/install_linter.sh install_linter.sh
39
COPY ./common/common_utils.sh common_utils.sh
40
RUN bash ./install_linter.sh
41
RUN rm install_linter.sh common_utils.sh
42

43
USER jenkins
44
CMD ["bash"]
45