add cudnn, magma, nccl

jfgrimm · Feb 1, 2024 · 969ef92 · 969ef92
1 parent 1ce12d9
commit 969ef92
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/easybuild/easyconfigs/p/PyTorch/PyTorch-2.1.2-foss-2023a-CUDA-12.1.1.eb b/easybuild/easyconfigs/p/PyTorch/PyTorch-2.1.2-foss-2023a-CUDA-12.1.1.eb
@@ -138,6 +138,9 @@ dependencies = [
     ('networkx', '3.1'),
     ('sympy', '1.12'),
     ('Z3', '4.12.2', '-Python-%(pyver)s'),
+    ('cuDNN', '8.9.2.26', versionsuffix, SYSTEM),
+    ('magma', '2.7.2', versionsuffix),
+    ('NCCL', '2.18.3', versionsuffix),
 ]
 
 excluded_tests = {