mikemhenry/cme-lab-images:cuda91

$ singularity pull shub://mikemhenry/cme-lab-images:cuda91

Singularity Recipe

Bootstrap: shub
From: shub://singularity-hub.org/mikemhenry/cme-lab-images:base

%help
Base cuda8 image used in CME-Lab workflows.

http://coen.boisestate.edu/cmelab/

%setup

%files

%labels
  Maintainer Mike Henry
  Maintaine_rEmail mikehenry@boisestate.edu
  Version v0.01

%environment

  PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
  LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64

  # nvidia-container-runtime
  NVIDIA_VISIBLE_DEVICES=all
  NVIDIA_DRIVER_CAPABILITIES=compute,utility
  NVIDIA_REQUIRE_CUDA="cuda>=9.1"

  # devel envs
  LIBRARY_PATH=/usr/local/cuda/lib64/stubs


  export PATH LD_LIBRARY_PATH NVIDIA_VISIBLE_DEVICES NVIDIA_DRIVER_CAPABILITIES NVIDIA_REQUIRE_CUDA LIBRARY_PATH



%post
  # Base https://gitlab.com/nvidia/cuda/blob/ubuntu16.04/9.2/base/Dockerfile
  export CUDA_VERSION=9.1.85
  export CUDA_PKG_VERSION=9-1=$CUDA_VERSION-1
  
  apt-get update && apt-get install -y --no-install-recommends ca-certificates apt-transport-https gnupg-curl
  rm -rf /var/lib/apt/lists/*
    
  NVIDIA_GPGKEY_SUM=d1be581509378368edeec8c1eb2958702feedf3bc3d17011adbf24efacce4ab5
  NVIDIA_GPGKEY_FPR=ae09fe4bbd223a84b2ccfce3f60f4b3d7fa2af80
  
  apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
  apt-key adv --export --no-emit-version -a $NVIDIA_GPGKEY_FPR | tail -n +5 > cudasign.pub
  echo "$NVIDIA_GPGKEY_SUM  cudasign.pub" | sha256sum -c --strict - && rm cudasign.pub 
  echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/cuda.list 
  echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list

  apt-get update && apt-get install -y --no-install-recommends cuda-cudart-$CUDA_PKG_VERSION
  ln -s cuda-9.1 /usr/local/cuda
  rm -rf /var/lib/apt/lists/*

  echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf
  echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf

  # Runtime https://gitlab.com/nvidia/cuda/blob/ubuntu16.04/9.2/runtime/Dockerfile

  export NCCL_VERSION=2.2.12

  apt-get update && apt-get install -y --no-install-recommends \
        cuda-libraries-$CUDA_PKG_VERSION \
        cuda-nvtx-$CUDA_PKG_VERSION \
        libnccl2=$NCCL_VERSION-1+cuda9.1
  rm -rf /var/lib/apt/lists/*

  # Devel https://gitlab.com/nvidia/cuda/blob/ubuntu16.04/9.2/devel/Dockerfile

  apt-get update && apt-get install -y --no-install-recommends \
        cuda-libraries-dev-$CUDA_PKG_VERSION \
        cuda-nvml-dev-$CUDA_PKG_VERSION \
        cuda-minimal-build-$CUDA_PKG_VERSION \
        cuda-command-line-tools-$CUDA_PKG_VERSION \
        libnccl-dev=$NCCL_VERSION-1+cuda9.1
  rm -rf /var/lib/apt/lists/*

 
%runscript

%test

Collection

• Name: mikemhenry/cme-lab-images
• License: None

Metrics

Feedback

Was this page helpful?

Glad to hear it! Please tell us how we can improve.

Sorry to hear that. Please tell us how we can improve.