Add CUDA 8.0 on CentOS 6

Closes: #227

8 files changed, 95 insertions, 2 deletions

diff --git a/centos-6/cuda/8.0/devel/Dockerfile b/centos-6/cuda/8.0/devel/Dockerfile
new file mode 100644
index 0000000..2ee437b
--- /dev/null
+++ b/centos-6/cuda/8.0/devel/Dockerfile
@@ -0,0 +1,22 @@
+FROM cuda:8.0-runtime
+MAINTAINER NVIDIA CORPORATION <digits@nvidia.com>
+
+RUN yum install -y \
+        cuda-core-$CUDA_PKG_VERSION \
+        cuda-misc-headers-$CUDA_PKG_VERSION \
+        cuda-command-line-tools-$CUDA_PKG_VERSION \
+        cuda-license-$CUDA_PKG_VERSION \
+        cuda-nvrtc-dev-$CUDA_PKG_VERSION \
+        cuda-nvml-dev-$CUDA_PKG_VERSION \
+        cuda-nvgraph-dev-$CUDA_PKG_VERSION \
+        cuda-cusolver-dev-$CUDA_PKG_VERSION \
+        cuda-cublas-dev-$CUDA_PKG_VERSION \
+        cuda-cufft-dev-$CUDA_PKG_VERSION \
+        cuda-curand-dev-$CUDA_PKG_VERSION \
+        cuda-cusparse-dev-$CUDA_PKG_VERSION \
+        cuda-npp-dev-$CUDA_PKG_VERSION \
+        cuda-cudart-dev-$CUDA_PKG_VERSION \
+        cuda-driver-dev-$CUDA_PKG_VERSION && \
+    rm -rf /var/cache/yum/*
+
+ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs:${LIBRARY_PATH}
diff --git a/centos-6/cuda/8.0/devel/cudnn b/centos-6/cuda/8.0/devel/cudnn
new file mode 120000
index 0000000..a738859
--- /dev/null
+++ b/centos-6/cuda/8.0/devel/cudnn
@@ -0,0 +1 @@
+cudnn5
\ No newline at end of file
diff --git a/centos-6/cuda/8.0/devel/cudnn5/Dockerfile b/centos-6/cuda/8.0/devel/cudnn5/Dockerfile
new file mode 100644
index 0000000..4c6d5e7
--- /dev/null
+++ b/centos-6/cuda/8.0/devel/cudnn5/Dockerfile
@@ -0,0 +1,12 @@
+FROM cuda:8.0-devel
+MAINTAINER NVIDIA CORPORATION <digits@nvidia.com>
+
+ENV CUDNN_VERSION 5
+LABEL com.nvidia.cudnn.version="5"
+
+RUN CUDNN_DOWNLOAD_SUM=a87cb2df2e5e7cc0a05e266734e679ee1a2fadad6f06af82a76ed81a23b102c8 && \
+    curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz -O && \
+    echo "$CUDNN_DOWNLOAD_SUM  cudnn-8.0-linux-x64-v5.1.tgz" | sha256sum -c - && \
+    tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local && \
+    rm cudnn-8.0-linux-x64-v5.1.tgz && \
+    ldconfig
diff --git a/centos-6/cuda/8.0/runtime/Dockerfile b/centos-6/cuda/8.0/runtime/Dockerfile
new file mode 100644
index 0000000..5c508ea
--- /dev/null
+++ b/centos-6/cuda/8.0/runtime/Dockerfile
@@ -0,0 +1,37 @@
+FROM centos:6
+MAINTAINER NVIDIA CORPORATION <digits@nvidia.com>
+
+LABEL com.nvidia.volumes.needed="nvidia_driver"
+
+RUN NVIDIA_GPGKEY_SUM=d1be581509378368edeec8c1eb2958702feedf3bc3d17011adbf24efacce4ab5 && \
+    curl -fsSL http://developer.download.nvidia.com/compute/cuda/repos/rhel6/x86_64/7fa2af80.pub | sed '/^Version/d' > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA && \
+    echo "$NVIDIA_GPGKEY_SUM  /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA" | sha256sum -c -
+
+COPY cuda.repo /etc/yum.repos.d/cuda.repo
+
+ENV CUDA_VERSION 8.0
+LABEL com.nvidia.cuda.version="8.0"
+
+ENV CUDA_PKG_VERSION 8-0-8.0.44-1
+RUN yum install -y \
+        cuda-nvrtc-$CUDA_PKG_VERSION \
+        cuda-nvgraph-$CUDA_PKG_VERSION \
+        cuda-cusolver-$CUDA_PKG_VERSION \
+        cuda-cublas-$CUDA_PKG_VERSION \
+        cuda-cufft-$CUDA_PKG_VERSION \
+        cuda-curand-$CUDA_PKG_VERSION \
+        cuda-cusparse-$CUDA_PKG_VERSION \
+        cuda-npp-$CUDA_PKG_VERSION \
+        cuda-cudart-$CUDA_PKG_VERSION && \
+    ln -s cuda-$CUDA_VERSION /usr/local/cuda && \
+    rm -rf /var/cache/yum/*
+
+RUN echo "/usr/local/cuda/lib" >> /etc/ld.so.conf.d/cuda.conf && \
+    echo "/usr/local/cuda/lib64" >> /etc/ld.so.conf.d/cuda.conf && \
+    ldconfig
+
+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/nvidia/bin:/usr/local/cuda/bin:${PATH}
+ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}
diff --git a/centos-6/cuda/8.0/runtime/cuda.repo b/centos-6/cuda/8.0/runtime/cuda.repo
new file mode 100644
index 0000000..2097276
--- /dev/null
+++ b/centos-6/cuda/8.0/runtime/cuda.repo
@@ -0,0 +1,6 @@
+[cuda]
+name=cuda
+baseurl=http://developer.download.nvidia.com/compute/cuda/repos/rhel6/x86_64
+enabled=1
+gpgcheck=1
+gpgkey=file:///etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA
\ No newline at end of file
diff --git a/centos-6/cuda/8.0/runtime/cudnn b/centos-6/cuda/8.0/runtime/cudnn
new file mode 120000
index 0000000..a738859
--- /dev/null
+++ b/centos-6/cuda/8.0/runtime/cudnn
@@ -0,0 +1 @@
+cudnn5
\ No newline at end of file
diff --git a/centos-6/cuda/8.0/runtime/cudnn5/Dockerfile b/centos-6/cuda/8.0/runtime/cudnn5/Dockerfile
new file mode 100644
index 0000000..0266d32
--- /dev/null
+++ b/centos-6/cuda/8.0/runtime/cudnn5/Dockerfile
@@ -0,0 +1,12 @@
+FROM cuda:8.0-runtime
+MAINTAINER NVIDIA CORPORATION <digits@nvidia.com>
+
+ENV CUDNN_VERSION 5
+LABEL com.nvidia.cudnn.version="5"
+
+RUN CUDNN_DOWNLOAD_SUM=a87cb2df2e5e7cc0a05e266734e679ee1a2fadad6f06af82a76ed81a23b102c8 && \
+    curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz -O && \
+    echo "$CUDNN_DOWNLOAD_SUM  cudnn-8.0-linux-x64-v5.1.tgz" | sha256sum -c - && \
+    tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local --wildcards 'cuda/lib64/libcudnn.so.*' && \
+    rm cudnn-8.0-linux-x64-v5.1.tgz && \
+    ldconfig
diff --git a/centos-6/cuda/Makefile b/centos-6/cuda/Makefile
index 3cfe735..b7b248c 100644
--- a/centos-6/cuda/Makefile
+++ b/centos-6/cuda/Makefile
@@ -1,7 +1,9 @@
 # Copyright (c) 2016, NVIDIA CORPORATION. All rights reserved.
 
-CUDA_VERSIONS := 7.5
+CUDA_VERSIONS := 8.0 \
+                 7.5
 
-CUDNN_VERSIONS := 7.5-cudnn5
+CUDNN_VERSIONS := 8.0-cudnn5 \
+                  7.5-cudnn5
 
 include ../../mk/cuda.mk