Linux Docker 映像

为简化本地安装,Driverless AI 将作为 Docker 映像提供给以下系统组合:

主机操作系统

Docker 版本

主机架构

最小内存

Ubuntu 16.04 或更高版本

Docker CE

x86_64

64 GB

RHEL 或 CentOS 7.4 或更高版本

Docker CE

x86_64

64 GB

NVIDIA DGX 注册表

x86_64

请注意 :推荐 CUDA 11.2.2 或更高版本,同时 NVIDIA 驱动程序 >= 471.68 (仅 GPU)。

若需获得 GPU 支持等最佳性能,请使用 nvidia-docker。若仅需在不使用 GPU 的情况下获得较低性能的体验,则使用普通版本的 docker(具有相同的 docker 映像)即可。

这些安装步骤假设您已有 Driverless AI 的许可证密钥。请访问 https://www.h2o.ai/driverless-ai/,了解如何获取 Driverless AI 许可证密钥。获取密钥后,首次登录时系统将提示您将许可证密钥粘贴至 Driverless AI UI 中,或者您可以将其保存为 .sig 文件并放置于您将在安装过程中创建的 license 文件夹中。

注意,从 1.10 版本开始,DAI docker 映像使用内部 tini 运行,相当于使用 docker 中的 --init 。如果启动命令中同时启用了这两项,tini 将输出一条(无害的)警告消息。对于 GPU 用户,由于 GPU 需要 --pid=host 用于 nvml,从而将 tini 设为不使用 pid=1,因此它会显示警告消息(仍然无害)。

我们推荐 docker 启动命令中使用 --shm-size=256m 。但如果用户打算广泛构建 image auto model ,则推荐在 Driverless AI docker 命令中使用 --shm-size=2g