From 13df72e61b1441f42908575f2c6a288b39b6b5a7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andreas=20L=C3=A4ndle?=
 <969523+alaendle@users.noreply.github.com>
Date: Sat, 13 Jun 2026 18:39:24 +0200
Subject: [PATCH] Pre-install libtorch (#8039).

---
 docker/03-custom-install.sh | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/docker/03-custom-install.sh b/docker/03-custom-install.sh
index ea38f903..8227a0ca 100755
--- a/docker/03-custom-install.sh
+++ b/docker/03-custom-install.sh
@@ -65,3 +65,12 @@ Z3_VER=4.13.4
     && rm z3-${Z3_VER}-x64-glibc-2.35.zip \
     && ln -s /usr/local/z3-${Z3_VER}-x64-glibc-2.35/bin/z3 /usr/bin/z3
 )
+
+# Install libtorch
+LIBTORCH_VERSION=2.5.0 
+LIBTORCH_HOME=/usr/local/libtorch
+LIBTORCH_CUDA_VERSION=cpu
+LIBTORCH_SKIP_DOWNLOAD=1
+curl -OL https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-$LIBTORCH_VERSION%2Bcpu.zip
+unzip libtorch-shared-with-deps-$LIBTORCH_VERSION%2Bcpu.zip -d /usr/local/
+rm libtorch-shared-with-deps-$LIBTORCH_VERSION%2Bcpu.zip