安装#

先决条件#

  1. Linux x64。

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

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

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

NGC 容器中的 DALI#

DALI 预装在 TensorFlowPyTorchPaddlePaddle 容器中,这些容器位于 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-cudaXXXnvidia-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)由**外部组织**托管。