From 969ef926a01e26defccfdc4f41b086cbde86ee6d Mon Sep 17 00:00:00 2001
From: jfgrimm <jasper.grimm@york.ac.uk>
Date: Thu, 1 Feb 2024 15:45:56 +0000
Subject: [PATCH] add cudnn, magma, nccl

---
 .../p/PyTorch/PyTorch-2.1.2-foss-2023a-CUDA-12.1.1.eb          | 3 +++
 1 file changed, 3 insertions(+)

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
index 6ffe56d33ed..6b2ab3b4970 100644
--- 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 = {