FROM stitchem/stitchem-base:latest ENV CUDA=10.1.105-1 ENV CUDA_APT=10-1 ENV CUDA_SHORT=10.1 ENV CUDA_INSTALLER=cuda-repo-ubuntu1804_${CUDA}_amd64.deb RUN wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/${CUDA_INSTALLER} RUN dpkg -i ${CUDA_INSTALLER} RUN apt update -qq && apt install -y \ cuda-core-${CUDA_APT} \ cuda-cudart-dev-${CUDA_APT} \ cuda-cufft-dev-${CUDA_APT} \ cuda-nvml-dev-${CUDA_APT} ENV CUDA_HOME=/usr/local/cuda-${CUDA_SHORT} ENV PATH=${PATH}:${CUDA_HOME}/bin ENV LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${CUDA_HOME}/lib64