安装 cuDNN Python 前端#

您可以从 pip wheel 或源代码安装 cuDNN Python 前端。安装 cuDNN Python 前端后，您可以运行 pytest 命令来验证安装。

从 pip Wheel 安装 Python 前端#

下载与您的 Python 安装对应的 pip wheel。

pip install nvidia_cudnn_frontend

从源代码安装 Python 前端#

所需的最低 Python 版本为 3.6。Python 绑定编译需要一个开发包，可以通过运行以下命令安装

apt-get install python-dev

  1. 如果您需要 CUDA Toolkit 或 cuDNN 软件的自定义安装路径，请按如下方式设置环境变量

    软件

    环境变量

    CUDA Toolkit

    CUDAToolkit_ROOT

    cuDNN

    CUDNN_PATH

    如果您不设置这些环境变量，则用于下载前端 Python API 的命令将从默认系统路径获取 CUDA Toolkit 和 cuDNN。

  2. 从 GitHub 上的 NVIDIA cudnn-frontend 项目下载前端 Python API。

    pip install git+https://github.com/NVIDIA/cudnn-frontend.git

验证 Python 前端安装#

要测试您的安装是否成功，请运行以下命令

pytest test/python_fe

包含 Python 前端库#

要包含 Python 前端库，请运行

import cudnn
在本页中