Start Docker Bash Session
You can use the following helper script to start an interactive bash session with a given 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
We provide several pre-built images for doing quick exploration with TVM installed.
For example, you can run the following command to get
Then inside the docker container, you can type the following command to start the jupyter notebook
Check out https://hub.docker.com/r/tvmai/ to get the full list of available prebuilt images.
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
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
./docker/ci_build.sh ci_gpu make -C docs html
build golang test suite.
./docker/build.sh ci_cpu tests/scripts/task_golang.sh