From 77fd3749b2b51a0ae974a1eb18cd37eea8480c94 Mon Sep 17 00:00:00 2001 From: Thomas Roeblitz Date: Sun, 9 Jun 2024 20:06:34 +0200 Subject: [PATCH] add PyTorch-bundle with CUDA --- .../2023.06/eessi-2023.06-eb-4.9.1-2023a.yml | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/easystacks/pilot.nessi.no/2023.06/eessi-2023.06-eb-4.9.1-2023a.yml b/easystacks/pilot.nessi.no/2023.06/eessi-2023.06-eb-4.9.1-2023a.yml index 762f1a236f..903979da71 100644 --- a/easystacks/pilot.nessi.no/2023.06/eessi-2023.06-eb-4.9.1-2023a.yml +++ b/easystacks/pilot.nessi.no/2023.06/eessi-2023.06-eb-4.9.1-2023a.yml @@ -55,6 +55,26 @@ easyconfigs: - PyTorch-2.1.2-foss-2023a-CUDA-12.1.1.eb: options: cuda-compute-capabilities: 6.0,6.1,7.0,7.5,8.0,8.6,8.9,9.0 + # PyTorch-bundle-CUDA's dependencies without CUDA + - librosa-0.10.1-foss-2023a.eb + - NLTK-3.8.1-foss-2023a.eb + - parameterized-0.9.0-GCCcore-12.3.0.eb + - Scalene-1.5.26-GCCcore-12.3.0.eb + - scikit-image-0.22.0-foss-2023a.eb + - SentencePiece-0.2.0-GCC-12.3.0.eb: + # see https://github.com/easybuilders/easybuild-easyconfigs/pull/19987 + options: + from-pr: 19987 + - libmad-0.15.1b-GCCcore-12.3.0.eb: + # see https://github.com/easybuilders/easybuild-easyconfigs/pull/19987 + options: + from-pr: 19987 + - SoX-14.4.2-GCCcore-12.3.0.eb: + # see https://github.com/easybuilders/easybuild-easyconfigs/pull/19987 + options: + from-pr: 19987 + - tensorboard-2.15.1-gfbf-2023a.eb + - tqdm-4.66.1-GCCcore-12.3.0.eb - PyTorch-bundle-2.1.2-foss-2023a-CUDA-12.1.1.eb: # see https://github.com/easybuilders/easybuild-easyconfigs/pull/20484 options: