mirror of
https://github.com/commercialhaskell/stackage.git
synced 2026-06-28 09:15:47 +02:00
Keep LIBTORCH environment variables for building.
This commit is contained in:
parent
b4de2a3e68
commit
aa97d8c006
13
Dockerfile
13
Dockerfile
@ -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
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user