[Docker] Install PyTorch on cpu image (#14842)

install PyTorch on cpu image
apache · May 13, 2023 · e54bbc7 · e54bbc7
1 parent 3756b71
commit e54bbc7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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