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
。