Keep LIBTORCH environment variables for building.

This commit is contained in:
Andreas Ländle 2026-06-13 19:44:36 +02:00
parent b4de2a3e68
commit aa97d8c006
2 changed files with 9 additions and 8 deletions

View File

@ -3,16 +3,21 @@ FROM ubuntu:24.04
ENV HOME /home/stackage
ENV LANG en_US.UTF-8
ADD docker/01-build-server.sh /tmp/01-build-server.sh
COPY docker/01-build-server.sh /tmp/01-build-server.sh
RUN /tmp/01-build-server.sh && rm /tmp/01-build-server.sh
ADD docker/02-apt-get-install.sh /tmp/02-apt-get-install.sh
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
ADD docker/03-custom-install.sh /tmp/03-custom-install.sh
ENV LIBTORCH_VERSION 2.5.0
ENV LIBTORCH_HOME /usr/local/libtorch
ENV LIBTORCH_CUDA_VERSION cpu
ENV LIBTORCH_SKIP_DOWNLOAD 1
COPY docker/03-custom-install.sh /tmp/03-custom-install.sh
RUN /tmp/03-custom-install.sh && rm /tmp/03-custom-install.sh
ADD docker/04-cleanup.sh /tmp/04-cleanup.sh
COPY docker/04-cleanup.sh /tmp/04-cleanup.sh
RUN /tmp/04-cleanup.sh && rm /tmp/04-cleanup.sh
# Include file path

View File

@ -67,10 +67,6 @@ Z3_VER=4.13.4
)
# 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