博主未授权任何人或组织机构转载博主任何原创文章,感谢各位对原创的支持!
博主链接



nvidia 高级特性MIG详细介绍(二)

【Nvidia】nvidia 高级特性MIG详细介绍(二)-LMLPHP

一、CUDA提供的并发机制

       MIG对于CUDA应用程序来说基本上是透明的,因此CUDA的编程模型可以保持不变,以降低编程工作量。CUDA已经公开了多种在GPU上并行运行工作的技术,下面展示了这些技术与MIG的比较。请注意,StreamsMPS是CUDA编程模型的一部分,因此可以在GPU实例中使用。

       CUDA Streams是一种CUDA编程模型特性,在CUDA应用程序中,不同的工作可以提交到独立的队列,并由GPU独立处理。CUDA Streams只能在单个进程内使用,并且不能提供太多的隔离性——地址空间是共享的,SM是共享的,GPU内存带宽、cache和内存容量是共享的。最后,任何错误都会影响所有Streams和整个执行过程

       MPS(Multi-Process service)是CUDA的多进程服务。它允许协作的多进程应用程序共享GPU上的计算资源。它通常用于协作的MPI工作,但也被用于不相关应用程序共享GPU资源,同时也接受了这种解决方案带来的挑战。MPS目前不提供客户端之间的错误隔离,但是可以限制每个MPS客户端使用的流多处理器在所有SMs中的比例,但调度硬件仍然是共享的。内存带宽、cache和内存容量都在MPS客户端之间共享。

       MIG是NVIDIA gpu提供的新的并发形式,同时解决了其他CUDA技术在运行并行工作时的一些限制。


二、MIG设备的命名规则

       默认情况下,MIG设备由一个“GPU实例”和一个“计算实例”组成。下表突出显示了一种命名规则,通过其GPU实例的和其以(而不仅仅是其内存切片数)来标记一个MIG设备。

【Nvidia】nvidia 高级特性MIG详细介绍(二)-LMLPHP

       下表描述了A100-SXM4-40GB配置文件名称说明。对于A100-SXM4-80GB,配置文件名称将根据内存比例而变化——例如1g.10gb、2g.20gb、3g.40gb、4g.40gb、7g.80gb。

       根据用户的工作负载,每个GI可以进一步细分为多个CI。下表主要显示了在这种情况下MIG设备的名称。所示示例是对3g.20gb设备的细分,分成一组具有不同计算实例切片数的子设备。

【Nvidia】nvidia 高级特性MIG详细介绍(二)-LMLPHP

三、CUDA设备枚举

       CUDA应用程序通过使用MIG设备标识,将程序允许在指定的MIG设备上。CUDA 11/R450和CUDA 12/R525只支持单个MIG实例的枚举。换句话说,无论创建了多少个MIG设备(或使容器可用),单个CUDA进程只能枚举一个MIG设备

       CUDA应用程序将一个CI及其父GI视为一个CUDA设备。CUDA只能使用单个CI,如果有多个CI可见,将选择第一个可用的CI。总而言之,有两个约束条件:

  • CUDA只能枚举单个计算实例;
  • 如果枚举了任何其它GPU上的计算实例,CUDA将不会枚举非MIG GPU;

扩展了CUDA_VISIBLE_DEVICES 对MIG的支持。根据所使用的驱动程序版本,支持以下两种格式:

  • 在drivers >= R470(470.42.01+)下,每个MIG设备都分配了一个GPU UUID,格式如下MIG- <UUID>;
  • drivers > R470(例如R450和R460),通过指定CI和相应的父GI来枚举每个MIG设备。该格式遵循如下约定:MIG-<GPU-UUID>/<GPU instance ID>/<compute instance ID>;
$ nvidia-smi -L        

GPU 0: A100-SXM4-40GB (UUID: GPU-5d5ba0d6-d33d-2b2c-524d-9e3d8d2b8a77)
  MIG 1g.5gb      Device  0: (UUID: MIG-c6d4f1ef-42e4-5de3-91c7-45d71c87eb3f)
  MIG 1g.5gb      Device  1: (UUID: MIG-cba663e8-9bed-5b25-b243-5985ef7c9beb)
  MIG 1g.5gb      Device  2: (UUID: MIG-1e099852-3624-56c0-8064-c5db1211e44f)
  MIG 1g.5gb      Device  3: (UUID: MIG-8243111b-d4c4-587a-a96d-da04583b36e2)
  MIG 1g.5gb      Device  4: (UUID: MIG-169f1837-b996-59aa-9ed5-b0a3f99e88a6)
  MIG 1g.5gb      Device  5: (UUID: MIG-d5d0152c-e3f0-552c-abee-ebc0195e9f1d)
  MIG 1g.5gb      Device  6: (UUID: MIG-7df6b45c-a92d-5e09-8540-a6b389968c31)
GPU 1: A100-SXM4-40GB (UUID: GPU-0aa11ebd-627f-af3f-1a0d-4e1fd92fd7b0)
  MIG 2g.10gb     Device  0: (UUID: MIG-0c757cd7-e942-5726-a0b8-0e8fb7067135)
  MIG 2g.10gb     Device  1: (UUID: MIG-703fb6ed-3fa0-5e48-8e65-1c5bdcfe2202)
  MIG 2g.10gb     Device  2: (UUID: MIG-532453fc-0faa-5c3c-9709-a3fc2e76083d)

四、可以使用的MIG 配置文件

因为我使用的是A100,所以这里用了A100的MIG配置文件,如下所示:
【Nvidia】nvidia 高级特性MIG详细介绍(二)-LMLPHP

A100-SXM4-40GB支持的配置文件如下表所示。对于A100-SXM4-80GB,配置文件名称将根据内存比例而变化——例如1g.10gb、1g.10gb+me、1g.20gb、2g.20gb、3g.40gb、4g.40gb以及7g.80gb。

【Nvidia】nvidia 高级特性MIG详细介绍(二)-LMLPHP



感谢大家的阅读,这里是从善若水的博客!


【Nvidia】nvidia 高级特性MIG详细介绍(二)-LMLPHP

04-03 02:19