From 6b9c4b7b9cb1b8ba7331007dc0ca684d12a50157 Mon Sep 17 00:00:00 2001
From: Alex Domingo <alex.domingo.toro@vub.be>
Date: Thu, 30 Nov 2023 17:13:17 +0100
Subject: [PATCH 1/2] adding easyconfigs:
 segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb

---
 ...ls-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 easybuild/easyconfigs/s/segmentation-models-pytorch/segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb

diff --git a/easybuild/easyconfigs/s/segmentation-models-pytorch/segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb b/easybuild/easyconfigs/s/segmentation-models-pytorch/segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb
new file mode 100644
index 00000000000..8be6181619b
--- /dev/null
+++ b/easybuild/easyconfigs/s/segmentation-models-pytorch/segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb
@@ -0,0 +1,46 @@
+easyblock = 'PythonBundle'
+
+name = 'segmentation-models-pytorch'
+version = '0.3.3'
+versionsuffix = '-CUDA-%(cudaver)s'
+
+homepage = 'https://github.com/qubvel/segmentation_models.pytorch'
+description = "Python library with Neural Networks for Image Segmentation based on PyTorch."
+
+toolchain = {'name': 'foss', 'version': '2022a'}
+
+dependencies = [
+    ('CUDA', '11.7.0', '', SYSTEM),
+    ('Python', '3.10.4'),
+    ('PyTorch', '1.12.0', '-CUDA-%(cudaver)s'),
+    ('torchvision', '0.13.1', '-CUDA-%(cudaver)s'),
+    ('Safetensors', '0.3.1', '-CUDA-%(cudaver)s'),
+    ('tqdm', '4.64.0'),
+]
+
+use_pip = True
+
+exts_list = [
+    ('munch', '2.5.0', {
+        'checksums': ['2d735f6f24d4dba3417fa448cae40c6e896ec1fdab6cdb5e6510999758a4dbd2'],
+    }),
+    ('pretrainedmodels', '0.7.4', {
+        'checksums': ['7e77ead4619a3e11ab3c41982c8ad5b86edffe37c87fd2a37ec3c2cc6470b98a'],
+    }),
+    ('efficientnet_pytorch', '0.7.1', {
+        'checksums': ['00b9b261effce59d2d47aae2ad238c29a2a65175470f41ada7ecac439b7c1ee1'],
+    }),
+    ('huggingface_hub', '0.15.1', {
+        'checksums': ['a61b7d1a7769fe10119e730277c72ab99d95c48d86a3d6da3e9f3d0f632a4081'],
+    }),
+    ('timm', '0.9.2', {
+        'checksums': ['d0977cc5e02c69bda979fca8b52aa315a5f2cb64ebf8ad2c4631b1e452762c14'],
+    }),
+    ('segmentation_models_pytorch', version, {
+        'checksums': ['b3b21ab4cd26a6b2b9e7a6ed466ace6452eb26ed3c31ae491ea2d7cbb01e384b'],
+    }),
+]
+
+sanity_pip_check = True
+
+moduleclass = 'lib'

From 6a357cfa17a0487ff91e2bd4b2b82f17fde3be03 Mon Sep 17 00:00:00 2001
From: Alex Domingo <alex.domingo.toro@vub.be>
Date: Fri, 8 Dec 2023 16:43:25 +0100
Subject: [PATCH 2/2] use common versionsuffix in deps of
 segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb

---
 ...mentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/easybuild/easyconfigs/s/segmentation-models-pytorch/segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb b/easybuild/easyconfigs/s/segmentation-models-pytorch/segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb
index 8be6181619b..b82795350a9 100644
--- a/easybuild/easyconfigs/s/segmentation-models-pytorch/segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb
+++ b/easybuild/easyconfigs/s/segmentation-models-pytorch/segmentation-models-pytorch-0.3.3-foss-2022a-CUDA-11.7.0.eb
@@ -12,9 +12,9 @@ toolchain = {'name': 'foss', 'version': '2022a'}
 dependencies = [
     ('CUDA', '11.7.0', '', SYSTEM),
     ('Python', '3.10.4'),
-    ('PyTorch', '1.12.0', '-CUDA-%(cudaver)s'),
-    ('torchvision', '0.13.1', '-CUDA-%(cudaver)s'),
-    ('Safetensors', '0.3.1', '-CUDA-%(cudaver)s'),
+    ('PyTorch', '1.12.0', versionsuffix),
+    ('torchvision', '0.13.1', versionsuffix),
+    ('Safetensors', '0.3.1', versionsuffix),
     ('tqdm', '4.64.0'),
 ]