陈少文的博客
·
NVIDIA 环境变量配置
💡
原文中文，约1400字，阅读约需4分钟。
📝
内容提要
本文介绍了NVIDIA GPU的配置选项，包括可见GPU的指定、驱动程序能力、CUDA版本要求及动态链接库路径，帮助用户正确设置CUDA环境。
🎯
关键要点
- NVIDIA_VISIBLE_DEVICES用于指定程序可见的GPU设备，支持通过索引或UUID指定。
- NVIDIA_DRIVER_CAPABILITIES控制容器内安装的驱动程序库和二进制文件。
- NVIDIA_REQUIRE_CUDA用于指定所需的CUDA版本和驱动程序版本。
- NVIDIA_REQUIRE_DRIVER用于指定所需的驱动程序版本。
- NVIDIA_REQUIRE_BRAND用于指定所需的GPU品牌。
- NVIDIA_REQUIRE_ARCH用于指定所需的GPU架构。
- CUDA_DEVICE_ORDER按照PCI_BUS_ID顺序排列GPU设备。
- LD_LIBRARY_PATH指定动态链接库的路径，以确保运行时找到CUDA库文件。
- CUDNN_PATH指定cuDNN库的安装路径。
- CUDA_HOME用于构建CUDA应用时搜索CUDA库文件和工具的路径。
- CUDA_PATH是NVIDIA开发工具搜索CUDA库文件和工具的路径。
❓
延伸问答
如何指定程序可见的GPU设备？
可以通过设置NVIDIA_VISIBLE_DEVICES环境变量，使用逗号分隔的GPU索引或UUID来指定可见的GPU设备。
NVIDIA_DRIVER_CAPABILITIES的作用是什么？
NVIDIA_DRIVER_CAPABILITIES用于控制容器内安装的驱动程序库和二进制文件，确保所需功能可用。
如何指定所需的CUDA版本？
可以通过设置NVIDIA_REQUIRE_CUDA环境变量来指定所需的CUDA版本和驱动程序版本，例如NVIDIA_REQUIRE_CUDA = 'cuda>=11.0 driver>=450'。
LD_LIBRARY_PATH的作用是什么？
LD_LIBRARY_PATH用于指定动态链接库的路径，以确保运行时能够找到CUDA库文件。
如何设置CUDA_HOME环境变量？
CUDA_HOME用于构建CUDA应用时搜索CUDA库文件和工具的路径，通常设置为CUDA的安装目录，例如CUDA_HOME = /usr/local/cuda。
NVIDIA_REQUIRE_ARCH用于什么？
NVIDIA_REQUIRE_ARCH用于指定所需的GPU架构，例如'maxwell'、'pascal'等，以确保兼容性。
➡️
