Commit 3a126dde authored by Viacheslav Ostroukh's avatar Viacheslav Ostroukh 🚲
Browse files

add Dockerfile for CUDA research container

parent d02d37ea
Pipeline #10010 passed with stage
in 17 seconds
FROM quantumtinkerer/research:latest
USER root
WORKDIR /
# FIXME 9.2 is hardcoded along the Dockerfile
ENV CUDA_VERSION 9.2.88
ENV CUDA_PKG_VERSION 9-2=${CUDA_VERSION}-1
# For now CUDA is not supported on 18.04, experimental image of NVidia uses this
# one
RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1710/x86_64/7fa2af80.pub | apt-key add - \
&& echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1710/x86_64 /" > /etc/apt/sources.list.d/cuda.list \
&& apt-get update \
&& apt-get install -y --no-install-recommends \
cuda-libraries-$CUDA_PKG_VERSION \
cuda-libraries-dev-$CUDA_PKG_VERSION \
cuda-nvtx-$CUDA_PKG_VERSION \
cuda-minimal-build-$CUDA_PKG_VERSION \
cuda-command-line-tools-$CUDA_PKG_VERSION \
cuda-cudart-$CUDA_PKG_VERSION \
&& apt-get clean \
&& ln -s /usr/local/cuda-9.2 /usr/local/cuda \
&& rm -rf /var/lib/apt/lists/*
RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && \
echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
ENV PATH /usr/local/cuda/bin:${PATH}
ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
ENV NVIDIA_REQUIRE_CUDA "cuda>=9.2"
USER $NB_UID
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment