容器共享GPU时查看容器使用的GPU编号

分类: 文章 • 2024-07-02 08:43:34

对于单机多卡的 GPU 服务器，在做容器的 GPU 分配的时候，可以选择通过环境变量 NVIDIA_VISIBLE_DEVICES来指定 GPU 的索引或者 UUID。
因为在容器中看到的 GPU 索引都是从0开始的，如果想找到对应母机上的 GPU 卡，通过 UUID 又不太好确定。
容器共享GPU时查看容器使用的GPU编号