跳至内容

Junyi's Lab

TVM Dockerfile (CPU only)

写了一个 Dockerfile,可以一键部署 TVM 的 CPU 版本

映射 container 端口 22 即可使用 ssh,用户名密码都是 tvm

在 GitHub 查看 Dockerfile

构建指令:

docker build -t tvm_cpu:0.7.0-x86_64 .

或者你也可以直接拉取我的镜像:

docker pull dockerjunyi/tvm_cpu:0.7.0-x86_64

使用方法:

# -p 容器内 22 端口映射到本地 2222
# -i 交互模式运行容器
# -t 开启伪输入终端(与 -i 连用)

# 运行一个容器,将容器内 22 端口映射到本地 2222,容器设置为交互模式,开启伪输入终端,容器 image 叫做 dockerjunyi/tvm_cpu:0.7.0-x86_64,同时执行容器内的 /bin/bash
docker run -p 2222:22 -it dockerjunyi/tvm_cpu:0.7.0-x86_64 /bin/bash

当然,运行起来容器之后,如果不想使用 docker 的交互模式,你也可以通过映射出的 ssh 端口来访问