安装#

前提条件#

  1. Linux x64。

  2. **NVIDIA 驱动程序** 支持 CUDA 11.0 或更高版本（即 450.80.02 或更高版本的驱动程序）。

  3. **CUDA 工具包** - 对于基于 CUDA 12 的 DALI，工具包是动态链接的，需要安装。对于 CUDA 11 构建，它是可选的。

  4. [可选] 以下一个或多个深度学习框架

NGC 容器中的 DALI#

DALI 预装在 TensorFlow、 PyTorch 和 PaddlePaddle 容器中，这些容器位于 NVIDIA GPU Cloud 上。

pip - 正式发布#

nvidia-dali#

执行以下命令以安装指定 CUDA 版本的最新 DALI（请查看 支持矩阵 以查看您的平台是否受支持）

  • 对于 CUDA 11.0

pip install --extra-index-url https://pypi.nvidia.com --upgrade nvidia-dali-cuda110

或者只是

pip install nvidia-dali-cuda110

  • 对于 CUDA 12.0

pip install --extra-index-url https://pypi.nvidia.com --upgrade nvidia-dali-cuda120

或者只是

pip install nvidia-dali-cuda120

注意

CUDA 11.0 和 CUDA 12.0 构建使用增强兼容性的 CUDA 工具包。它使用最新的 CUDA 11.x/12.x 分别构建工具包，同时可以在最新的稳定 CUDA 11.0 和 CUDA 12.0 兼容驱动程序（分别为 450.80 或更高版本和 525.60 或更高版本）上运行。使用最新的驱动程序可能会启用其他功能。更多详细信息可以在 增强 CUDA 兼容性指南 中找到。

注意

请始终使用可用的最新版本的 pip（至少 >= 19.3），并在可能的情况下通过执行 pip install –upgrade pip 进行更新

nvidia-dali-tf-plugin#

DALI 不包含 DALI TensorFlow 插件的预构建版本。它需要作为单独的软件包安装，该软件包将针对当前安装的 TensorFlow 版本构建

  • 对于 CUDA 11.0

pip install --extra-index-url https://pypi.nvidia.com --upgrade nvidia-dali-tf-plugin-cuda110

或者只是

pip install nvidia-dali-tf-plugin-cuda110

  • 对于 CUDA 12.0

pip install --extra-index-url https://pypi.nvidia.com --upgrade nvidia-dali-tf-plugin-cuda120

或者只是

pip install nvidia-dali-tf-plugin-cuda120

如果尚未安装，安装此软件包将安装 nvidia-dali-cudaXXX 及其依赖项。 必须先安装软件包 tensorflow-gpu，然后才能尝试安装 nvidia-dali-tf-plugin-cudaXXX。

注意

软件包 nvidia-dali-tf-plugin-cudaXXX 和 nvidia-dali-cudaXXX 应为完全相同的版本。 因此，安装最新的 nvidia-dali-tf-plugin-cudaXXX 将替换已安装的任何旧版本的 nvidia-dali-cudaXXX。 要使用旧版本的 DALI，请在 pip install 命令中显式提供版本。

pip - 每晚构建和每周构建#

注意

虽然可从每晚和每周构建下载的二进制文件包含 GitHub 中可用的最新更改，但与正式版本相比，某些功能可能无法正常工作或提供较差的性能。 这些构建旨在供寻求最新可用版本并准备大胆探索前所未有领域的早期采用者使用。

注意

建议在安装每晚或每周构建之前卸载常规 DALI 和 TensorFlow 插件，因为它们安装在同一路径中

每晚构建#

要访问最新的每晚构建，请使用以下发布渠道

  • 对于 CUDA 11.0

pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly --upgrade nvidia-dali-nightly-cuda110
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly --upgrade nvidia-dali-tf-plugin-nightly-cuda110

  • 对于 CUDA 12.0

pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly --upgrade nvidia-dali-nightly-cuda120
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly --upgrade nvidia-dali-tf-plugin-nightly-cuda120

每周构建#

此外，还有一个经过更彻底测试的每周发布渠道。 要访问最新的每周构建，请使用以下发布渠道（仅适用于 CUDA 12）

pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/weekly --upgrade nvidia-dali-weekly-cuda120
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/weekly --upgrade nvidia-dali-tf-plugin-weekly-cuda120

pip - 旧版本发布#

对于旧版本的 DALI（0.22 及更低版本），请使用软件包 nvidia-dali。 可以通过更改 pip 索引来选择 CUDA 版本

pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/9.0 --upgrade nvidia-dali
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/9.0 --upgrade nvidia-dali-tf-plugin

pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/10.0 --upgrade nvidia-dali
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/10.0 --upgrade nvidia-dali-tf-plugin

pip install --extra-index-url https://developer.download.nvidia.com/compute/redist --upgrade nvidia-dali-cuda102
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist --upgrade nvidia-dali-tf-plugin-cuda102

pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/11.0 --upgrade nvidia-dali
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/11.0 --upgrade nvidia-dali-tf-plugin

从 DALI 0.22.0 开始提供 CUDA 11 构建。

从 DALI 1.4.0 到 DALI 1.20 提供 CUDA 10.2 构建。

到 DALI 1.3.0 提供 CUDA 10 构建。

到 DALI 0.22.0 提供 CUDA 9 构建。

开放认知环境 (Open-CE)#

DALI 也可作为开放认知环境的一部分提供 - 该项目包含为机器学习和深度学习框架集合构建 conda 软件包所需的一切。

这项工作由社区驱动，此处提供的 DALI 版本可能不是最新的。

预构建软件包（包括 DALI）由 **外部组织** 托管。