Google Git
Sign in
apache / tvm / refs/tags/v0.7.0 / . / docker
tree: 4c7ac815fc7443a7e96d9a104b7122949e5007cf [path history] [tgz]
  1. install/
  2. bash.sh
  3. build.sh
  4. dev_common.sh
  5. Dockerfile.ci_cpu
  6. Dockerfile.ci_gpu
  7. Dockerfile.ci_i386
  8. Dockerfile.ci_jekyll
  9. Dockerfile.ci_lint
  10. Dockerfile.ci_qemu
  11. Dockerfile.ci_wasm
  12. Dockerfile.conda_cpu
  13. Dockerfile.conda_cuda100
  14. Dockerfile.conda_cuda90
  15. Dockerfile.demo_android
  16. Dockerfile.demo_cpu
  17. Dockerfile.demo_gpu
  18. Dockerfile.demo_opencl
  19. Dockerfile.demo_vitis_ai
  20. lint.sh
  21. README.md
  22. with_the_same_user
docker/README.md

TVM Docker

This directory contains the TVM's docker infrastructure. We use docker to provide build environments for CI and images for demo. We need docker and nvidia-docker for GPU images.

Start Docker Bash Session

You can use the following helper script to start an interactive bash session with a given image_name.

/path/to/tvm/docker/bash.sh image_name

The script does the following things:

  • Mount current directory to /workspace and set it as home
  • Switch user to be the same user that calls the bash.sh
  • Use the host-side network

The helper bash script can be useful to build demo sessions.

Prebuilt Docker Images

You can use third party pre-built images for doing quick exploration with TVM installed. For example, you can run the following command to launch tvmai/demo-cpu image.

/path/to/tvm/docker/bash.sh tvmai/demo-cpu

Then inside the docker container, you can type the following command to start the jupyter notebook

jupyter notebook

You can find some un-official prebuilt images in https://hub.docker.com/r/tvmai/ . Note that these are convenience images and are not part of the ASF release.

Use Local Build Script

We also provide script to build docker images locally. We use (build.sh)[./build.sh] to build and run the commands. To build and run docker images, we can run the following command at the root of the project.

./docker/build.sh image_name [command]

Here image_name corresponds to the docker defined in the Dockerfile.image_name.

You can also start an interactive session by typing

./docker/build.sh image_name -it bash

The build command will map the tvm root to /workspace/ inside the container with the same user as the user invoking the docker command. Here are some common use examples to perform CI tasks.

  • lint the python codes

    ./docker/build.sh ci_lint make pylint

  • build codes with CUDA support

    ./docker/build.sh ci_gpu make -j$(nproc)

  • do the python unittest

    ./docker/build.sh ci_gpu tests/scripts/task_python_unittest.sh

  • build the documents. The results will be available at docs/_build/html

    ./docker/ci_build.sh ci_gpu make -C docs html

  • build golang test suite.

    ./docker/build.sh ci_cpu tests/scripts/task_golang.sh