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
No related branches found
No related tags found
No related merge requests found
Bootstrap: shub Bootstrap: oras
From: mkandes/naked-singularity:ubuntu-18.04 From: ghcr.io/mkandes/naked-singularity:ubuntu-18.04
%labels %labels
...@@ -10,19 +10,12 @@ From: mkandes/naked-singularity:ubuntu-18.04 ...@@ -10,19 +10,12 @@ From: mkandes/naked-singularity:ubuntu-18.04
AUTHOR_NAME Marty Kandes AUTHOR_NAME Marty Kandes
AUTHOR_EMAIL mkandes@sdsc.edu AUTHOR_EMAIL mkandes@sdsc.edu
LAST_UPDATED 20210427 LAST_UPDATED 20210903
%setup %setup
%environment %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 # Set NVIDIA driver and CUDA versions
export CUDA_DRIVER_VERSION='460' export CUDA_DRIVER_VERSION='460'
export CUDA_MAJOR='10' export CUDA_MAJOR='10'
...@@ -30,7 +23,7 @@ From: mkandes/naked-singularity:ubuntu-18.04 ...@@ -30,7 +23,7 @@ From: mkandes/naked-singularity:ubuntu-18.04
export CUDA_VERSION="${CUDA_MAJOR}.${CUDA_MINOR}" export CUDA_VERSION="${CUDA_MAJOR}.${CUDA_MINOR}"
export CUDNN_MAJOR='8' export CUDNN_MAJOR='8'
export CUDNN_MINOR='1' export CUDNN_MINOR='1'
export CUDNN_REVISION='0.77' export CUDNN_REVISION='1.33'
export CUDNN_VERSION="${CUDNN_MAJOR}.${CUDNN_MINOR}.${CUDNN_REVISION}" export CUDNN_VERSION="${CUDNN_MAJOR}.${CUDNN_MINOR}.${CUDNN_REVISION}"
export NCCL_MAJOR='2' export NCCL_MAJOR='2'
export NCCL_MINOR='8' export NCCL_MINOR='8'
...@@ -73,7 +66,7 @@ From: mkandes/naked-singularity:ubuntu-18.04 ...@@ -73,7 +66,7 @@ From: mkandes/naked-singularity:ubuntu-18.04
export CUDA_VERSION="${CUDA_MAJOR}.${CUDA_MINOR}" export CUDA_VERSION="${CUDA_MAJOR}.${CUDA_MINOR}"
export CUDNN_MAJOR='8' export CUDNN_MAJOR='8'
export CUDNN_MINOR='1' export CUDNN_MINOR='1'
export CUDNN_REVISION='0.77' export CUDNN_REVISION='1.33'
export CUDNN_VERSION="${CUDNN_MAJOR}.${CUDNN_MINOR}.${CUDNN_REVISION}" export CUDNN_VERSION="${CUDNN_MAJOR}.${CUDNN_MINOR}.${CUDNN_REVISION}"
export NCCL_MAJOR='2' export NCCL_MAJOR='2'
export NCCL_MINOR='8' export NCCL_MINOR='8'
...@@ -126,11 +119,6 @@ From: mkandes/naked-singularity:ubuntu-18.04 ...@@ -126,11 +119,6 @@ From: mkandes/naked-singularity:ubuntu-18.04
apt-get -y libglu1-mesa apt-get -y libglu1-mesa
apt-get -y libglu1-mesa-dev 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 # Cleanup
apt-get -y autoremove --purge apt-get -y autoremove --purge
apt-get -y clean apt-get -y clean
......
...@@ -23,7 +23,7 @@ From: ghcr.io/mkandes/naked-singularity:ubuntu-18.04 ...@@ -23,7 +23,7 @@ From: ghcr.io/mkandes/naked-singularity:ubuntu-18.04
export CUDA_VERSION="${CUDA_MAJOR}.${CUDA_MINOR}" export CUDA_VERSION="${CUDA_MAJOR}.${CUDA_MINOR}"
export CUDNN_MAJOR='8' export CUDNN_MAJOR='8'
export CUDNN_MINOR='1' export CUDNN_MINOR='1'
export CUDNN_REVISION='0.77' export CUDNN_REVISION='1.33'
export CUDNN_VERSION="${CUDNN_MAJOR}.${CUDNN_MINOR}.${CUDNN_REVISION}" export CUDNN_VERSION="${CUDNN_MAJOR}.${CUDNN_MINOR}.${CUDNN_REVISION}"
export NCCL_MAJOR='2' export NCCL_MAJOR='2'
export NCCL_MINOR='8' 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