Skip to content

Commit

Permalink
add PyTorch-bundle with CUDA
Browse files Browse the repository at this point in the history
  • Loading branch information
truib committed Jun 9, 2024
1 parent 30d22b4 commit 77fd374
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions easystacks/pilot.nessi.no/2023.06/eessi-2023.06-eb-4.9.1-2023a.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 77fd374

Please sign in to comment.