FROM registry.yandex.net/maps/core-base-bionic:9754240

# avoid apt-get errors
ENV DEBIAN_FRONTEND noninteractive

ENV nvidia_version 455.32.00
ENV nvidia_version_major 455
ENV cuda_version 10.1.243-1
ENV cuda_version_code 10-1
ENV cuda_version_major_minor 10.1
ENV nvidia_repo "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64"
ENV cuda_conf=/etc/ld.so.conf.d/cuda-10.1.conf

RUN bash -c 'set -e; \
    apt-get update; \
    mkdir -p /usr/lib/nvidia; \
    for package in \
            "libnvidia-compute-${nvidia_version_major}_${nvidia_version}-0ubuntu1_amd64.deb"; do \
        wget --quiet ${nvidia_repo}/${package};  \
        apt install --yes --fix-broken --no-install-recommends ./$package; \
        rm -f ${package}; \
    done; \
    echo /usr/lib/x86_64-linux-gnu >> "${cuda_conf}"; \
    echo /usr/lib/nvidia-${nvidia_version_major} > /etc/ld.so.conf.d/nvidia-${nvidia_version_major}.conf; \
    ldconfig; '

COPY install /
