adding easyconfigs: Kornia-0.7.1-foss-2023a-CUDA-12.1.1.eb

Author: Samuel Moors

easybuild/easyconfigs/k/Kornia/Kornia-0.7.1-foss-2023a-CUDA-12.1.1.eb

@@ -0,0 +1,42 @@
+easyblock = 'CargoPythonBundle'
+
+name = 'Kornia'
+version = '0.7.1'
+versionsuffix = '-CUDA-12.1.1'
+
+homepage = 'https://kornia.github.io/'
+description = """
+Kornia is a differentiable computer vision library that provides a rich set of
+differentiable image processing and geometric vision algorithms. Built on top
+of PyTorch, Kornia integrates seamlessly into existing AI workflows, allowing
+you to leverage powerful batch transformations, auto-differentiation and GPU
+acceleration. Whether you’re working on image transformations, augmentations,
+or AI-driven image processing, Kornia equips you with the tools you need to
+bring your ideas to life."""
+
+toolchain = {'name': 'foss', 'version': '2023a'}
+
+builddependencies = [
+    ('maturin', '1.1.0'),
+]
+dependencies = [
+    ('Python', '3.11.3'),
+    ('PyTorch', '2.1.2', versionsuffix),
+]
+
+_whl_tmpl = '%(name)s-%(version)s-cp311-cp311-manylinux_2_17_{}.manylinux2014_{}.whl'
+
+exts_list = [
+    ('kornia_rs', '0.1.9', {
+        'sources': [_whl_tmpl.format('%(arch)s', '%(arch)s')],
+        'checksums': [{
+            _whl_tmpl.format('x86_64', 'x86_64'): '6cdda9133297c4cff2c2c54be44d5c39bce715306d0bccb8ab1fae7c0dc7cf63',
+            _whl_tmpl.format('aarch64', 'aarch64'): '54604bc8eb7d4d703eac19963378b4d6a72432a3f0da765edb1b0396d10def01',
+        }],
+    }),
+    ('kornia', version, {
+        'checksums': ['65b54a50f70c1f88240b557fda3fdcc1ab866982a5d062e52213130f5a48465c'],
+    }),
+]
+
+moduleclass = 'vis'