Skip to content
Snippets Groups Projects
Commit 8b9ec6f9 authored by Marty Kandes's avatar Marty Kandes
Browse files

Also update CUDNN_REVISION in Singularity.ubuntu-18.04-cuda-10.2

parent 09c6bbd8
Branches
No related tags found
No related merge requests found
Bootstrap: shub
From: mkandes/naked-singularity:ubuntu-18.04
Bootstrap: oras
From: ghcr.io/mkandes/naked-singularity:ubuntu-18.04
%labels
......@@ -10,19 +10,12 @@ From: mkandes/naked-singularity:ubuntu-18.04
AUTHOR_NAME Marty Kandes
AUTHOR_EMAIL mkandes@sdsc.edu
LAST_UPDATED 20210427
LAST_UPDATED 20210903
%setup
%environment
# Location of NVIDIA/CUDA repositories
export CUDA_ROOT_URL='http://developer.download.nvidia.com/compute'
export CUDA_DISTRO='ubuntu1804'
export CUDA_ARCH='x86_64'
export CUDA_REPO_URL="${CUDA_ROOT_URL}/cuda/repos/${CUDA_DISTRO}/${CUDA_ARCH}"
export CUDA_GPG_PUBKEY='7fa2af80.pub'
# Set NVIDIA driver and CUDA versions
export CUDA_DRIVER_VERSION='460'
export CUDA_MAJOR='10'
......@@ -30,7 +23,7 @@ From: mkandes/naked-singularity:ubuntu-18.04
export CUDA_VERSION="${CUDA_MAJOR}.${CUDA_MINOR}"
export CUDNN_MAJOR='8'
export CUDNN_MINOR='1'
export CUDNN_REVISION='0.77'
export CUDNN_REVISION='1.33'
export CUDNN_VERSION="${CUDNN_MAJOR}.${CUDNN_MINOR}.${CUDNN_REVISION}"
export NCCL_MAJOR='2'
export NCCL_MINOR='8'
......@@ -73,7 +66,7 @@ From: mkandes/naked-singularity:ubuntu-18.04
export CUDA_VERSION="${CUDA_MAJOR}.${CUDA_MINOR}"
export CUDNN_MAJOR='8'
export CUDNN_MINOR='1'
export CUDNN_REVISION='0.77'
export CUDNN_REVISION='1.33'
export CUDNN_VERSION="${CUDNN_MAJOR}.${CUDNN_MINOR}.${CUDNN_REVISION}"
export NCCL_MAJOR='2'
export NCCL_MINOR='8'
......@@ -126,11 +119,6 @@ From: mkandes/naked-singularity:ubuntu-18.04
apt-get -y libglu1-mesa
apt-get -y libglu1-mesa-dev
# Set paths to CUDA binaries and libraries
export CUDA_HOME="/usr/local/cuda-${CUDA_VERSION}"
export PATH="${CUDA_HOME}/bin${PATH:+:${PATH}}"
export LD_LIBRARY_PATH="${CUDA_HOME}/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}"
# Cleanup
apt-get -y autoremove --purge
apt-get -y clean
......
......@@ -23,7 +23,7 @@ From: ghcr.io/mkandes/naked-singularity:ubuntu-18.04
export CUDA_VERSION="${CUDA_MAJOR}.${CUDA_MINOR}"
export CUDNN_MAJOR='8'
export CUDNN_MINOR='1'
export CUDNN_REVISION='0.77'
export CUDNN_REVISION='1.33'
export CUDNN_VERSION="${CUDNN_MAJOR}.${CUDNN_MINOR}.${CUDNN_REVISION}"
export NCCL_MAJOR='2'
export NCCL_MINOR='8'
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment