跳至主要内容

CUDA with NCCL/cuDNN

  • 套件官網
  • release note
  • 套件來源:由於 NVIDIA NGC CUDA 容器已同時安裝 NCCL/cuDNN,故已此為參考,統一複製容器內搭配的檔案
  • 除此之外,NVIDIA HPC SDK 也已包含 CUDA 和 NCCL/OpenMPI,但缺少 cuDNN
  • 提醒 CUDA 11 以後,部分 library 有可能無法搭配和 CentOS 7 預設提供的 gcc 4.8.5,可參考 release note 支援表,搭配 gcc 7 以上的 module 使用 ref: https://docs.nvidia.com/hpc-sdk/hpc-sdk-release-notes/index.html

CUDA with NCCL/cuDNN module matrix

module nameusage
cuda/11.3ml cuda/11.3
cuda/11.2ml cuda/11.2
cuda/11.1ml cuda/11.1
cuda/11.0ml cuda/11.0
cuda/10.2ml cuda/10.2
cuda/10.1ml cuda/10.1
cuda/10.0ml cuda/10.0
cuda/9.2ml cuda/9.2
cuda/9.0ml cuda/9.0

CUDA with NCCL/cuDNN matrix

CUDAw/ NCCLw/ cuDNNbased on CUDA container
11.32.9.98.2.011.3.1-cudnn8-devel-centos7
11.22.8.48.1.111.2.2-cudnn8-devel-centos7
11.12.8.48.0.511.1.1-cudnn8-devel-centos7
11.02.9.68.0.511.0.3-cudnn8-devel-centos7
10.22.9.68.2.010.2-cudnn8-devel-centos7
10.12.8.47.6.510.1-cudnn7-devel-centos7
10.02.6.47.6.510.0-cudnn7-devel-centos7
9.2N/A7.6.59.2-cudnn7-devel-centos7
9.0N/A7.6.59.0-cudnn7-devel-centos7