对接组合中两种GPU访问方式的区别

Question

我遇到了两种让容器访问docker-compose文件中的GPU的方法：

    environment:      
      NVIDIA_VISIBLE_DEVICES: <device>

   deploy:
       resources:
         reservations:
           devices:
             - driver: nvidia
               count: 1
               capabilities: [gpu]

如果有什么不同，这两者有什么区别呢？互联网上的简短搜索没有发现任何东西。

Answer 1

我觉得这份文件说得很清楚：https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/user-guide.html

这一部分：

NVIDIA容器工具包为枚举GPU和CUDA容器所支持的功能提供了不同的选项。本用户指南演示了NVIDIA容器工具包的下列功能：

将NVIDIA运行时注册为自定义运行时到Docker

使用环境变量启用以下功能：

枚举GPU并控制哪些GPU对容器可见

使用功能控制驱动程序的哪些功能对容器可见。

使用约束控制运行时的行为

第二个方法是手动定义要使容器可用的主机设备。我相信NVIDIA正试图通过提供第一个环境变量方法来使您的生活更轻松。我相信，如果第二条路径正确地设置了所有所需的字段，这两种方法都会奏效。