交叉编译 cuDNN 示例#

本节介绍如何在 Ubuntu 上交叉编译 cuDNN 示例。

Linux arm64-SBSA#

按照本节中的步骤在 Linux arm64-SBSA 上交叉编译 cuDNN 示例。 Linux arm64-SBSA 采用基于 ARM 的 CPU 核心,用于服务器基础系统架构 (SBSA)。

安装适用于 Linux arm64-SBSA 的 CUDA 工具包#

按照 Linux 版 CUDA 安装指南 中描述的说明安装适用于工具包交叉 SBSA 变体的 CUDA 工具包。

安装适用于 Linux arm64-SBSA 的 cuDNN#

  1. 按照 在 Linux 上安装 cuDNN 中描述的说明安装适用于库交叉 SBSA 变体的 cuDNN。

  2. 安装 arm64-SBSA 主机编译器。

    sudo apt-get -y install g++-aarch64-linux-gnu
    

交叉编译适用于 Linux arm64-SBSA 的 cuDNN 示例#

  1. cudnn_samples_v9 目录复制到您的主目录。

    cp -r /usr/src/cudnn_samples_v9 $HOME
    
  2. 对于每个示例,执行以下命令。

    cd $HOME/cudnn_samples_v9/(each sample)
    sudo make TARGET_ARCH=aarch64 SBSA=1
    

Linux aarch64-Jetson#

按照本节中的步骤在 Ubuntu 上交叉编译 cuDNN 示例。 Linux aarch64-Jetson 采用基于 ARM 的 CPU 核心,用于 Tegra 平台。

安装适用于 Linux aarch64-Jetson 的 CUDA 工具包#

按照 Linux 版 CUDA 安装指南 中描述的说明安装适用于工具包交叉 Jetson 变体的 CUDA 工具包。

安装适用于 Linux aarch64-Jetson 的 cuDNN#

按照 在 Linux 上安装 cuDNN 中描述的说明安装适用于库交叉 Jetson 变体的 cuDNN。

交叉编译适用于 Linux aarch64-Jetson 的 cuDNN 示例#

  1. cudnn_samples_v9 目录复制到您的主目录。

    cp -r /usr/src/cudnn_samples_v9 $HOME
    
  2. 对于每个示例,执行以下命令。

    cd $HOME/cudnn_samples_v9/(each sample)
    sudo make TARGET_ARCH=aarch64