Next try for libtorch-ffi dependencies (#8039).

This commit is contained in:
Andreas Ländle 2026-06-14 10:30:01 +02:00
parent 87e76e946d
commit a7771b2b41
2 changed files with 4 additions and 3 deletions

View File

@ -10,7 +10,6 @@ COPY docker/02-apt-get-install.sh /tmp/02-apt-get-install.sh
RUN /tmp/02-apt-get-install.sh && rm /tmp/02-apt-get-install.sh
ENV LIBTORCH_VERSION 2.5.0
ENV LIBTORCH_HOME /usr/local/libtorch
ENV LIBTORCH_CUDA_VERSION cpu
ENV LIBTORCH_SKIP_DOWNLOAD 1
@ -21,4 +20,4 @@ COPY docker/04-cleanup.sh /tmp/04-cleanup.sh
RUN /tmp/04-cleanup.sh && rm /tmp/04-cleanup.sh
# Include file path
ENV CPATH /usr/lib/jvm/java-8-openjdk-amd64/include:/usr/lib/jvm/java-8-openjdk-amd64/include/linux:/usr/lib/llvm-3.7/include:/usr/local/libtorch/include
ENV CPATH /usr/lib/jvm/java-8-openjdk-amd64/include:/usr/lib/jvm/java-8-openjdk-amd64/include/linux:/usr/lib/llvm-3.7/include

View File

@ -69,5 +69,7 @@ Z3_VER=4.13.4
# Install libtorch
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/
mv /usr/local/libtorch/lib/* /usr/lib/
mv /usr/local/libtorch/include/* /usr/include/
rm libtorch-shared-with-deps-$LIBTORCH_VERSION%2Bcpu.zip
mv /usr/local/libtorch/lib/lib* /usr/lib/
rm -rf /usr/local/libtorch