Docker 镜像

我们提供 Docker 实用工具脚本来帮助开发者设置开发环境。它们也有助于运行 TVM 演示和教程。如果我们要使用 CUDA，我们需要 docker 和 nvidia-docker。

获取 tvm 源代码发行版或克隆 github 仓库以获取辅助脚本

git clone --recursive https://github.com/apache/tvm tvm

然后我们可以使用以下命令来启动 Docker 镜像。

/path/to/tvm/docker/bash.sh <image-name>

这里的 image-name 可以是本地 Docker 镜像名称，例如 tvm.ci_cpu，在你完成本地构建之后。

此辅助脚本执行以下操作

• 将当前目录挂载到 /workspace

• 切换用户为调用 bash.sh 的同一用户 (以便你可以读/写宿主机系统)

• 在 Linux 上使用宿主机侧网络。在 macOS 上使用桥接网络并暴露端口 8888，因为不支持宿主机网络驱动程序。（以便你可以使用 jupyter notebook）

然后你可以通过输入以下命令来启动 Jupyter notebook

jupyter notebook

在 macOS 上启动 Jupyter notebook 时，你可能会看到错误 OSError: [Errno 99] Cannot assign requested address。你可以通过以下方式更改绑定 IP 地址

jupyter notebook --ip=0.0.0.0

请注意，在 macOS 上，由于 bash.sh 使用 Docker 桥接网络，Jupyter 据报告将在类似 http://{container_hostname}:8888/?token=... 的 URL 上运行。在将其粘贴到浏览器中时，你应该将 container_hostname 替换为 localhost。

Docker 源代码

如果你有兴趣构建自己的 Docker 镜像，请查看 Docker 源代码。

运行以下命令来构建 Docker 镜像。

/path/to/tvm/docker/build.sh <image-name>

你可以在 https://hub.docker.com/r/tlcpack/ 找到一些非官方的第三方预构建镜像。这些镜像用于测试目的，并非 ASF 发布版本。