From e54bbc73a7bc9d942fa8d4794a4c2a6ff78a6421 Mon Sep 17 00:00:00 2001 From: masahi Date: Sun, 14 May 2023 03:27:14 +0900 Subject: [PATCH] [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 18d86b61fc3a..208a0f272a00 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