在容器和GPU中使用GPU,并且在Google Cloud VM中使用容器优化的操作系统

问题描述 投票:0回答:1

我想在Google Compute Engine上使用GPU运行自定义Docker映像。我已经构建了映像并将其推送到Google Container Registry。在Google Cloud Engine的主机上使用容器优化的OS似乎是合乎逻辑的,因为除了Docker,Nvidia GPU驱动程序和nvidia-container-runtime,我不需要在主机上安装任何其他软件。

我设法用this solution安装nvidia驱动程序。

但是如果没有nvidia-container运行时,我无法使用GPU(使用--gpu all选项)运行Docker映像。 This step is specified in official Docker documentation

是否可以在Google Cloud VM的容器优化的OS上安装nvidia-container-runtime?

google-cloud-platform gpu google-compute-engine google-container-os
1个回答
0
投票

您不必设置--gpu all,因为这是nvidia-container-runtime的默认选项。没有运行时,它将无法通过容器访问物理硬件的驱动程序。这个问题还不是很清楚,我只能说运行时周围没有办法,运行时充当物理硬件的抽象和驱动程序之间的粘合剂。该场景与类型2虚拟机管理程序非常相似,后者需要操作系统安装驱动程序,因为它无法通过绕过内核直接访问CPU或GPU。为了更好地了解虚拟CPU,只需研究虚拟CPU的实际工作方式即可。

© www.soinside.com 2019 - 2024. All rights reserved.