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