diff --git a/Dockerfile b/Dockerfile index b049c6d3..1f91c166 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,7 @@ FROM ubuntu:24.04 -ENV HOME /home/stackage -ENV LANG en_US.UTF-8 +ENV HOME=/home/stackage +ENV LANG=en_US.UTF-8 COPY docker/01-build-server.sh /tmp/01-build-server.sh RUN /tmp/01-build-server.sh && rm /tmp/01-build-server.sh @@ -9,9 +9,9 @@ RUN /tmp/01-build-server.sh && rm /tmp/01-build-server.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 -ENV LIBTORCH_VERSION 2.5.0 -ENV LIBTORCH_CUDA_VERSION cpu -ENV LIBTORCH_SKIP_DOWNLOAD 1 +ENV LIBTORCH_VERSION=2.5.0 +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 @@ -20,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 +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 diff --git a/docker/03-custom-install.sh b/docker/03-custom-install.sh index d5ac93cb..4eb2669d 100755 --- a/docker/03-custom-install.sh +++ b/docker/03-custom-install.sh @@ -69,7 +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/ +cp -rf /usr/local/libtorch/lib/* /usr/lib/ +cp -rf /usr/local/libtorch/include/* /usr/include/ rm libtorch-shared-with-deps-$LIBTORCH_VERSION%2Bcpu.zip rm -rf /usr/local/libtorch