NVIDIA_VISIBLE_DEVICES

指定程序可见的 GPU 设备

1

CUDA_VISIBLE_DEVICES=0,1

可用值:

• 1,2，以逗号分隔的 GPU UUID 或索引列表
• all，所有 GPU
• none，加载驱动，但无法访问 GPU
• void，不加载驱动

NVIDIA_DRIVER_CAPABILITIES

控制哪些驱动程序库/二进制文件将被安装在容器内

1

NVIDIA_DRIVER_CAPABILITIES=compute,utility

可用值:

• compute，CUDA 和 OpenCL 应用程序所需。
• compat32，运行 32 位应用程序所需。
• graphics，运行 OpenGL 和 Vulkan 应用程序所需。
• utility，nvidia-smi 使用和 NVML 所需。
• video，使用视频编解码器 SDK 所必需的。
• display，利用 X11 显示所需。

NVIDIA_REQUIRE_CUDA

用于指定所需的 CUDA 版本和驱动程序版本

1

NVIDIA_REQUIRE_CUDA="cuda>=11.0 driver>=450"

1

NVIDIA_REQUIRE_CUDA="cuda>=11.7"

NVIDIA_REQUIRE_DRIVER

用于指定所需的驱动程序版本

1

NVIDIA_REQUIRE_DRIVER="driver>=470"

NVIDIA_REQUIRE_BRAND

用于指定所需的 GPU 品牌

1
2

NVIDIA_REQUIRE_BRAND="tesla"
NVIDIA_REQUIRE_BRAND="geforce"

NVIDIA_REQUIRE_ARCH

用于指定所需的 GPU 架构

1

NVIDIA_REQUIRE_ARCH "maxwell pascal volta turing ampere"

CUDA_DEVICE_ORDER

按照 PCI_BUS_ID 顺序从 0 开始排列 GPU 设备。

1

export CUDA_VISIBLE_DEVICES='PCI_BUS_ID'

只使用前两个 GPU

LD_LIBRARY_PATH

指定动态链接库的路径，确保运行时可以找到 CUDA 库文件。

1

LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/lib/x86_64-linux-gnu:/usr/local/nvidia/lib

CUDNN_PATH

指定 cuDNN 库的安装路径

1

CUDNN_PATH=/usr/local/cuda

CUDA_HOME

构建 CUDA 应用时，搜索 CUDA 库文件和工具的路径

1

CUDA_HOME=/usr/local/cuda

CUDA_PATH

NVIDIA 自己的开发工具搜索 CUDA 库文件和工具的路径

1

CUDA_PATH=/usr/local/cuda