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

对于单机多卡的 GPU 服务器,在做容器的 GPU 分配的时候,可以选择通过环境变量 NVIDIA_VISIBLE_DEVICES来指定 GPU 的索引或者 UUID。 因为在容器中看到的 GPU 索引都是从0开始的,如果想找到对应母机上的 GPU 卡,通过 UUID 又不太好确定。
相关文章
相关标签/搜索