Skip to content

Instantly share code, notes, and snippets.

@mkolod

mkolod/build_your_own_cuda_docker_image.txt

Created June 7, 2020 00:22
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save mkolod/3aee141207d8bcbf405a12a75caab1b5 to your computer and use it in GitHub Desktop.
Save mkolod/3aee141207d8bcbf405a12a75caab1b5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
build_your_own_cuda_docker_image.txt
RUN wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
RUN chmod +x cuda_10.2.89_440.33.01_linux.run && \
./cuda_10.2.89_440.33.01_linux.run --silent --toolkit
# After installing CUDA, you need to add the env vars below, otherwise you'll get CUDA runtime version/driver
# version issue, even if you're using nvidia-docker.
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment