Skip to content

S
stitchEm
Commit 408d4f10 authored by Clement Guedez's avatar Clement Guedez Committed by jeremad
Browse files
Options

libnvidia ml on ubuntu 18.04

parent 9945f44b
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
docker/base_cuda_10.dockerfile
View file @ 408d4f10
...@@ -4,14 +4,16 @@ ENV CUDA=10.1.105-1 ...@@ -4,14 +4,16 @@ ENV CUDA=10.1.105-1
ENV CUDA_APT=10-1 ENV CUDA_APT=10-1
ENV CUDA_SHORT=10.1 ENV CUDA_SHORT=10.1
ENV CUDA_INSTALLER=cuda-repo-ubuntu1804_${CUDA}_amd64.deb ENV CUDA_INSTALLER=cuda-repo-ubuntu1804_${CUDA}_amd64.deb
ENV NV_LIB=libnvidia-compute-418
RUN wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/${CUDA_INSTALLER} RUN wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/${CUDA_INSTALLER}
RUN dpkg -i ${CUDA_INSTALLER} RUN dpkg -i ${CUDA_INSTALLER}
RUN apt update -qq && apt install -y \ RUN apt update -qq && apt install -y \
cuda-core-${CUDA_APT} \ cuda-compiler-${CUDA_APT} \
cuda-cudart-dev-${CUDA_APT} \ cuda-cudart-dev-${CUDA_APT} \
cuda-cufft-dev-${CUDA_APT} \ cuda-cufft-dev-${CUDA_APT} \
cuda-nvml-dev-${CUDA_APT} cuda-nvml-dev-${CUDA_APT} \
${NV_LIB}
ENV CUDA_HOME=/usr/local/cuda-${CUDA_SHORT} ENV CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
ENV PATH=${PATH}:${CUDA_HOME}/bin ENV PATH=${PATH}:${CUDA_HOME}/bin
......
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