示例#

示例演示了 Python 和 C++ 前端 API 的用法。

Python 示例#

Python 示例是 Jupyter 笔记本，其中包含使用前端 API 的分步说明。

源代码#

GitHub 上的 Python 接口示例。

依赖项#

要运行 Python 示例，您需要 requirements.txt 中提到的依赖项。这可以通过运行以下命令安装

pip install -r requirements.txt

C++ 示例#

源代码#

GitHub 上的 C++ 接口示例。

构建#

以下编译步骤仅在构建 C++ 示例时是必需的。

  1. 根据 https://cmake.com.cn/cmake/help/latest/module/FindCUDAToolkit.html 提供 CUDA 安装路径。

  2. 使用 CUDNN_PATH 环境变量或 CMake 参数提供 cuDNN 安装路径。CUDNN_PATH 包括 cuDNN 安装

    • 头文件位于 CUDNN_PATH/include 中。

    • 库文件位于 CUDNN_PATH/lib、CUDNN_PATH/lib64 或 CUDNN_PATH/lib/x64 中。

  3. 对于内部源构建：运行

    mkdir build
cd build
cmake -DCUDNN_PATH=/path/to/cudnn -DCUDAToolkit_ROOT=/path/to/cuda  ../
cmake --build . -j16
bin/samples

要跳过构建示例，请使用 -DCUDNN_FRONTEND_BUILD_SAMPLES=OFF。

要跳过构建 python 绑定，请使用 -DCUDNN_FRONTEND_BUILD_PYTHON_BINDINGS=OFF。

要添加调试符号，请使用 -DCMAKE_BUILD_TYPE=Debug。

如果您有陈旧的 CMake 缓存并想要更新 cuDNN 和 CUDA 路径，请删除 CMake 缓存（或构建目录并重复上述步骤）。

在本页中