容器共享GPU時查看容器使用的GPU編號

對於單機多卡的 GPU 服務器,在做容器的 GPU 分配的時候,可以選擇通過環境變量 NVIDIA_VISIBLE_DEVICES來指定 GPU 的索引或者 UUID。 因爲在容器中看到的 GPU 索引都是從0開始的,如果想找到對應母機上的 GPU 卡,通過 UUID 又不太好確定。
相關文章
相關標籤/搜索