矩池云上安装cuDNN头文件教程

简介

NVIDIA CUDA® 深度神经网络库 (cuDNN，NVIDIA CUDA Deep Neural Network) 是经 GPU 加速的深度神经网络基元库。cuDNN 可大幅优化标准例程（例如用于前向传播和反向传播的卷积层、池化层、归一化层和激活层）的实施。

世界各地的深度学习研究人员和框架开发者都依赖 cuDNN 实现高性能 GPU 加速。借助 cuDNN，研究人员和开发者可以专注于训练神经网络及开发软件应用，而不必花时间进行低层级的 GPU 性能调整。cuDNN 可加速广泛应用的深度学习框架，包括 Caffe2、Chainer、Keras、MATLAB、MxNet、PyTorch 和 TensorFlow。

cuDNN官网：https://developer.nvidia.com/zh-cn/cudnn

安装

NVIDIA官方GitLab地址：https://gitlab.com/nvidia/container-images/cuda/

安装脚本在dist目录下

选择cuda的版本号

矩池云上cuda8/cuda9使用的是ubuntu16.04，cuda10及以上都是ubuntu18.04

  • base：cuda安装方式
  • devel：nvcc等安装方式
  • runtime：cudnn等安装方式

检查系统版本

source /etc/os-release && echo $VERSION_ID

导入apt国内镜像仓库和其GPG key

# 第一步若输出 16.04，请用下面的命令
curl -fsSL https://mirrors.cloud.tencent.com/nvidia-machine-learning/ubuntu1604/x86_64/7fa2af80.pub | apt-key add -
echo "deb https://mirrors.cloud.tencent.com/nvidia-machine-learning/ubuntu1604/x86_64/ /" > /etc/apt/sources.list.d/cuda.list

# 第一步若输出 18.04，请用下面的命令
curl -fsSL https://mirrors.cloud.tencent.com/nvidia-machine-learning/ubuntu1804/x86_64/7fa2af80.pub | apt-key add -
echo "deb https://mirrors.cloud.tencent.com/nvidia-machine-learning/ubuntu1804/x86_64/ /" > /etc/apt/sources.list.d/cuda.list

cuDNN在腾讯源的nvidia-machine-learning中有，软件的对应的版本可以在腾讯源中搜索查到。

安装所需版本的 cudnn

以cuda10.2下cudnn 8.0.3版本为例，根据Dockerfile得出安装命令如下

# 这里以cudnn 8.0.3.33版本为例
apt-get update && apt-get install -y --no-install-recommends \
    libcudnn8=8.0.3.33-1+cuda10.2 \
    && apt-mark hold libcudnn8 && \
    rm -rf /var/lib/apt/lists/*
nvidia
cudnn
客服