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