Skip to content

Commit

Permalink
add compatibility notes for cuda/9.2 version
Browse files Browse the repository at this point in the history
  • Loading branch information
fgeorgatos authored and jirikuncar committed Feb 22, 2019
1 parent 3cae2e9 commit 5ed1f46
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion docker/cuda-9.2/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ FROM $BASE_IMAGE

MAINTAINER Swiss Data Science Center <[email protected]>

# Insall cuda
# Install cuda
# From: https://gitlab.com/nvidia/cuda/blob/ubuntu18.04/9.2/base/Dockerfile
USER root
RUN apt-get update && apt-get install -y --no-install-recommends gnupg2 curl ca-certificates && \
Expand All @@ -13,6 +13,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends gnupg2 curl ca-
apt-get purge --autoremove -y curl && \
rm -rf /var/lib/apt/lists/*

# Compatible with NVIDIA graphics driver R396.37 and higher ; known to work with cuDNN v7.1.2
ENV CUDA_VERSION 9.2.148

ENV CUDA_PKG_VERSION 9-2=$CUDA_VERSION-1
Expand Down

0 comments on commit 5ed1f46

Please sign in to comment.