Skip to content

Commit

Permalink
nuke cugraph (needs closed source repo, can't compile from source
Browse files Browse the repository at this point in the history
apparently)
  • Loading branch information
paulgessinger committed Jul 11, 2024
1 parent 0fb3693 commit 969b6f7
Showing 1 changed file with 0 additions and 16 deletions.
16 changes: 0 additions & 16 deletions ubuntu2204_exatrkx/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -60,13 +60,11 @@ ENV XXHASH_VERSION=0.7.3
ENV JSON_VERSION=3.11.2
ENV ROOT_VERSION=6.28.06
ENV TORCH_SCATTER_VERSION=2.1.2
ENV CUGRAPH_VERSION=22.02.00
ENV ONNXRUNTIME_VERSION=1.13.1

# libtorch (unzip cannot be used in a pipe...)
ENV LIBTORCH_URL_GPU https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.3.1%2Bcu121.zip


# xxHash
RUN mkdir -p src \
&& ${GET} https://github.com/Cyan4973/xxHash/archive/v${XXHASH_VERSION}.tar.gz \
Expand Down Expand Up @@ -129,20 +127,6 @@ RUN mkdir src \
&& cmake --build build -- install \
&& rm -rf build src

# cugraph
RUN mkdir src \
&& ${GET} https://github.com/rapidsai/cugraph/archive/refs/tags/v${CUGRAPH_VERSION}.tar.gz \
| ${UNPACK_TO_SRC} \
&& cmake -B build -S src/cpp -GNinja \
-DCMAKE_BUILD_TYPE=MinSizeRel \
-DCMAKE_CXX_STANDARD=20 \
-DCMAKE_INSTALL_PREFIX=${PREFIX} \
-DCMAKE_CUDA_ARCHITECTURES=${CUDA_ARCH} \
-DBUILD_TESTS=OFF \
-DBUILD_CUGRAPH_MG_TESTS=OFF \
&& cmake --build build -- install \
&& rm -rf build src

# Onnx (download of tar.gz does not work out of the box, since the build.sh script requires a git repository)
RUN git clone https://github.com/microsoft/onnxruntime src \
&& (cd src && git checkout v${ONNXRUNTIME_VERSION}) \
Expand Down

0 comments on commit 969b6f7

Please sign in to comment.