This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new e54bbc73a7 [Docker] Install PyTorch on cpu image (#14842)
e54bbc73a7 is described below
commit e54bbc73a7bc9d942fa8d4794a4c2a6ff78a6421
Author: masahi <[email protected]>
AuthorDate: Sun May 14 03:27:14 2023 +0900
[Docker] Install PyTorch on cpu image (#14842)
install PyTorch on cpu image
---
docker/Dockerfile.ci_cpu | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/docker/Dockerfile.ci_cpu b/docker/Dockerfile.ci_cpu
index 18d86b61fc..208a0f272a 100644
--- a/docker/Dockerfile.ci_cpu
+++ b/docker/Dockerfile.ci_cpu
@@ -145,3 +145,7 @@ ENV PATH /opt/sccache:$PATH
# Libxsmm deps
COPY install/ubuntu_install_libxsmm.sh /install
RUN bash /install/ubuntu_install_libxsmm.sh
+
+# ONNX and PyTorch
+COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
+RUN bash /install/ubuntu_install_onnx.sh
