{lib}[foss/2022a] DeepLabCut v2.3.6 with CUDA 11.7.0

easybuild/easyconfigs/d/DeepLabCut/DeepLabCut-2.3.6-foss-2022a-CUDA-11.7.0.eb

@@ -0,0 +1,74 @@
+# Loosely based on PR #7680 
+# J. Sassmannshausen (Imperial College London/UK)
+# upgrade to version 2.3.6: J Hein, (LUNARC, Lund University, Sweden)
+
+easyblock = 'PythonBundle'
+
+name = 'DeepLabCut'
+version = '2.3.6'
+versionsuffix = '-CUDA-%(cudaver)s'
+local_version = '%(versionsuffix)s-contrib'
+
+homepage = 'http://www.mousemotorlab.org/deeplabcut'
+description = "Markerless tracking of user-defined features with deep learning"
+
+toolchain = {'name': 'foss', 'version': '2022a'}
+toolchainopts = {'pic': True}
+
+dependencies = [
+    ('CUDA', '11.7.0', '', SYSTEM),
+    ('Python', '3.10.4'),
+    ('TensorFlow', '2.11.0',  versionsuffix),
+    ('PyTorch', '1.12.0', versionsuffix),
+    ('scikit-image', '0.19.3'),
+    ('matplotlib', '3.5.2'),
+    ('numba', '0.56.4'),
+    ('scikit-learn', '1.1.2'),
+    ('FFmpeg', '4.4.2'),
+    ('PyZMQ', '24.0.1'),
+    ('tqdm', '4.64.0'),
+    ('imageio', '2.22.2'),
+    ('statsmodels', '0.13.1'),
+    ('PyTables', '3.8.0'),
+    ('ruamel.yaml', '0.17.21'),
+    ('imgaug', '0.4.0', versionsuffix),
+]
+
+sanity_pip_check = True
+use_pip = True
+
+exts_list = [
+    ('dlclibrary', '0.0.4', {
+        'checksums': ['2b4ca078dc9cddb2a1f30d42cbc9d5c0e849e0f93f2b6781ca478baed0a829b8'],
+    }),
+    ('filterpy', '1.4.5', {
+        'sources': ['%(name)s-%(version)s.zip'],
+        'checksums': ['4f2a4d39e4ea601b9ab42b2db08b5918a9538c168cff1c6895ae26646f3d73b1'],
+    }),
+    ('huggingface-hub', '0.17.3', {
+        'sources': ['huggingface_hub-%(version)s.tar.gz'],
+        'checksums': ['40439632b211311f788964602bf8b0d9d6b7a2314fba4e8d67b2ce3ecea0e3fd'],
+    }),
+    ('imageio-ffmpeg', '0.4.9', {
+        'checksums': ['39bcd1660118ef360fa4047456501071364661aa9d9021d3d26c58f1ee2081f5'],
+    }),
+    ('msgpack-numpy', '0.4.8', {
+        'sources': ['msgpack-numpy-0.4.8.tar.gz'],
+        'checksums': ['c667d3180513422f9c7545be5eec5d296dcbb357e06f72ed39cc683797556e69'],
+    }),
+    ('tensorpack', '0.11', {
+        'checksums': ['022b610e416e62e3575424cd08e60af27808a5fb6914294615391caf582cbd4f'],
+    }),
+    ('tf_slim', '1.1.0', {
+        'source_urls': ['https://github.com/google-research/tf-slim/archive'],
+        'sources': ['v%(version)s.tar.gz'],
+        'checksums': ['964cde4b7728a408dcd5c841ab6b93d95137ab4b60db28b10400f86286bfeb8b'],
+    }),
+    (name, version, {
+        'source_tmpl': '%(namelower)s-%(version)s.tar.gz',
+        'checksums': ['c4479f05b3c917a6198c1abbb6ea0ac580ce4b02fe1d068e0215b1984aa8e385'],
+    }),
+]
+
+
+moduleclass = 'lib'