Skip to content

Commit e9683c5

Browse files
committed
Revert "Drop libtorch"
This reverts commit 2571dbc.
1 parent 04e8867 commit e9683c5

2 files changed

Lines changed: 22 additions & 0 deletions

File tree

ubuntu2404_gnn/Dockerfile

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,3 +47,12 @@ RUN apt-get update -y \
4747
cmake \
4848
bc \
4949
&& apt-get clean -y
50+
51+
# manual builds for hep-specific packages
52+
ENV GET curl --location --silent --create-dirs
53+
ENV UNPACK_TO_SRC tar -xz --strip-components=1 --directory src
54+
ENV PREFIX /usr/local
55+
ENV CUDA_ARCH 86
56+
57+
RUN ${GET} https://download.pytorch.org/libtorch/cu128/libtorch-cxx11-abi-shared-with-deps-2.7.1%2Bcu128.zip \
58+
| bsdtar -xz --strip-components=1 --directory /usr/local

ubuntu2404_tensorrt/Dockerfile

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,3 +49,16 @@ RUN apt-get update -y \
4949
time \
5050
bc \
5151
&& apt-get clean -y
52+
53+
# manual builds for hep-specific packages
54+
ENV GET="curl --location --silent --create-dirs"
55+
ENV PREFIX=/usr/local
56+
57+
# libtorch (unzip cannot be used in a pipe...)
58+
ENV LIBTORCH_URL_GPU=https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.3.1%2Bcu121.zip
59+
60+
RUN ${GET} https://github.com/Kitware/CMake/releases/download/v3.30.0/cmake-3.30.0-Linux-x86_64.tar.gz \
61+
| tar -xz --strip-components=1 --directory ${PREFIX}
62+
63+
RUN ${GET} https://acts.web.cern.ch/ACTS/ci/exatrkx/libtorch-cxx11-abi-shared-with-deps-2.3.1+cu121.tar.zst \
64+
| tar -I zstd -x --strip-components=1 --directory ${PREFIX}

0 commit comments

Comments
 (0)