Skip to content

Commit

Permalink
add comment
Browse files Browse the repository at this point in the history
  • Loading branch information
matbun committed Jan 10, 2025
1 parent 2f57939 commit f0ffb33
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions env-files/torch/createEnvVega.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,15 @@
#!/bin/bash
# -*- coding: utf-8 -*-

# --------------------------------------------------------------------------------------
# Part of the interTwin Project: https://www.intertwin.eu/
#
# Created by: Matteo Bunino
#
# Credit:
# - Matteo Bunino <[email protected]> - CERN
# --------------------------------------------------------------------------------------

if [ ! -f "env-files/torch/generic_torch.sh" ]; then
echo "ERROR: env-files/torch/generic_torch.sh not found!"
exit 1
Expand All @@ -13,10 +22,8 @@ ml Python/3.11.5-GCCcore-13.2.0
ml CMake/3.24.3-GCCcore-11.3.0
ml mpi4py
ml OpenMPI
#ml CUDA/11.7
ml CUDA/12.3
ml GCCcore/11.3.0
#ml NCCL/2.12.12-GCCcore-11.3.0-CUDA-11.7.0
ml NCCL
ml cuDNN/8.9.7.29-CUDA-12.3.0
ml UCX-CUDA/1.15.0-GCCcore-13.2.0-CUDA-12.3.0
Expand Down

0 comments on commit f0ffb33

Please sign in to comment.