{vis}[foss/2024a] torchvision v0.21.0 w/ CUDA 12.6.0

[smoors]

(created using eb --new-pr)

draft because PyTorch CUDA is not merged yet and we might switch to v2.7

[github-actions]

Updated software torchvision-0.21.0-foss-2024a-CUDA-12.6.0.eb

Diff against torchvision-0.21.0-foss-2024a.eb

easybuild/easyconfigs/t/torchvision/torchvision-0.21.0-foss-2024a.eb

diff --git a/easybuild/easyconfigs/t/torchvision/torchvision-0.21.0-foss-2024a.eb b/easybuild/easyconfigs/t/torchvision/torchvision-0.21.0-foss-2024a-CUDA-12.6.0.eb
index 12b38f33bd..014544d812 100644
--- a/easybuild/easyconfigs/t/torchvision/torchvision-0.21.0-foss-2024a.eb
+++ b/easybuild/easyconfigs/t/torchvision/torchvision-0.21.0-foss-2024a-CUDA-12.6.0.eb
@@ -2,6 +2,7 @@ easyblock = 'PythonBundle'
 
 name = 'torchvision'
 version = '0.21.0'
+versionsuffix = '-CUDA-12.6.0'
 
 homepage = 'https://github.com/pytorch/vision'
 description = " Datasets, Transforms and Models specific to Computer Vision"
@@ -19,7 +20,7 @@ dependencies = [
     ('SciPy-bundle', '2024.05'),
     ('FFmpeg', '7.0.2'),
     ('Pillow', '10.4.0'),
-    ('PyTorch', '2.6.0'),
+    ('PyTorch', '2.6.0', versionsuffix),
     ('libwebp', '1.4.0'),
 ]
 
@@ -28,7 +29,7 @@ exts_list = [
         'source_tmpl': 'pytest_mock-%(version)s.tar.gz',
         'checksums': ['159e9edac4c451ce77a5cdb9fc5d1100708d2dd4ba3c3df572f14097351af80e'],
     }),
-    (name, version, {
+    ('torchvision', '0.21.0', {
         'source_urls': ['https://github.com/pytorch/vision/archive'],
         'sources': [{'download_filename': 'v%(version)s.tar.gz', 'filename': '%(name)s-%(version)s.tar.gz'}],
         'patches': [

Diff against torchvision-0.17.0-foss-2023b.eb

easybuild/easyconfigs/t/torchvision/torchvision-0.17.0-foss-2023b.eb

diff --git a/easybuild/easyconfigs/t/torchvision/torchvision-0.17.0-foss-2023b.eb b/easybuild/easyconfigs/t/torchvision/torchvision-0.21.0-foss-2024a-CUDA-12.6.0.eb
index 464e1b2be1..014544d812 100644
--- a/easybuild/easyconfigs/t/torchvision/torchvision-0.17.0-foss-2023b.eb
+++ b/easybuild/easyconfigs/t/torchvision/torchvision-0.21.0-foss-2024a-CUDA-12.6.0.eb
@@ -1,36 +1,57 @@
+easyblock = 'PythonBundle'
+
 name = 'torchvision'
-version = '0.17.0'
+version = '0.21.0'
+versionsuffix = '-CUDA-12.6.0'
 
 homepage = 'https://github.com/pytorch/vision'
 description = " Datasets, Transforms and Models specific to Computer Vision"
 
-toolchain = {'name': 'foss', 'version': '2023b'}
+toolchain = {'name': 'foss', 'version': '2024a'}
 toolchainopts = {'cstd': 'c++17'}
 
-source_urls = ['https://github.com/pytorch/vision/archive']
-sources = ['v%(version)s.tar.gz']
-patches = [
-    '%(name)s-%(version)s_fix-build-with-FFmpeg6.patch',
-    '%(name)s-%(version)s_quantized_tol.patch'
-]
-checksums = [
-    {'v0.17.0.tar.gz': '55e395d5c7d9bf7658c82ac633cac2224aa168e1bfe8bb5b2b2a296c792a3500'},
-    {'torchvision-0.17.0_fix-build-with-FFmpeg6.patch':
-     '64404865a489c0348f79684419831ebd71ece7c7326a80184785578a368047b5'},
-    {'torchvision-0.17.0_quantized_tol.patch': '991b959f7d5e0fd418a163fa3ad828a2b60a64d0ce894cf579847138b13b62af'},
-]
-
 builddependencies = [
-    ('CMake', '3.27.6'),
+    ('CMake', '3.29.3'),
 ]
 
 dependencies = [
-    ('Python', '3.11.5'),
-    ('Python-bundle-PyPI', '2023.10'),
-    ('SciPy-bundle', '2023.11'),
-    ('FFmpeg', '6.0'),
-    ('Pillow', '10.2.0'),
-    ('PyTorch', '2.1.2'),
+    ('Python', '3.12.3'),
+    ('Python-bundle-PyPI', '2024.06'),
+    ('SciPy-bundle', '2024.05'),
+    ('FFmpeg', '7.0.2'),
+    ('Pillow', '10.4.0'),
+    ('PyTorch', '2.6.0', versionsuffix),
+    ('libwebp', '1.4.0'),
+]
+
+exts_list = [
+    ('pytest-mock', '3.14.1', {
+        'source_tmpl': 'pytest_mock-%(version)s.tar.gz',
+        'checksums': ['159e9edac4c451ce77a5cdb9fc5d1100708d2dd4ba3c3df572f14097351af80e'],
+    }),
+    ('torchvision', '0.21.0', {
+        'source_urls': ['https://github.com/pytorch/vision/archive'],
+        'sources': [{'download_filename': 'v%(version)s.tar.gz', 'filename': '%(name)s-%(version)s.tar.gz'}],
+        'patches': [
+            'torchvision-0.17.0_quantized_tol.patch',
+        ],
+        'installopts': '-v',
+        'runtest': (
+            'pytest'
+            ' -m "not xfail"'  # don't run tests that are expected that they might fail
+            ' -k "not test_frame_reading_mem_vs_file"'  # this one hangs
+            '" and not test_antialias_default_warning"'  # flaky test which is removed in later versions
+            '" and not test_videoapi"'  # video support is unstable
+            '" and not avif and not heic"'  # requires torchvision-extra-decoders, libavif PyTorch support, libheic
+            '" and not test_pretrained_pos and not test_equivalent_behavior_weights"'  # tests out of date?
+        ),
+        'testinstall': True,
+        'checksums': [
+            {'torchvision-0.21.0.tar.gz': '0a4a967bbb7f9810f792cd0289a07fb98c8fb5d1303fae8b63e3a6b05d720058'},
+            {'torchvision-0.17.0_quantized_tol.patch':
+             '991b959f7d5e0fd418a163fa3ad828a2b60a64d0ce894cf579847138b13b62af'},
+        ],
+    }),
 ]
 
 moduleclass = 'vis'

Diff against torchvision-0.16.0-foss-2023a-CUDA-12.1.1.eb

easybuild/easyconfigs/t/torchvision/torchvision-0.16.0-foss-2023a-CUDA-12.1.1.eb

diff --git a/easybuild/easyconfigs/t/torchvision/torchvision-0.16.0-foss-2023a-CUDA-12.1.1.eb b/easybuild/easyconfigs/t/torchvision/torchvision-0.21.0-foss-2024a-CUDA-12.6.0.eb
index 68696ef7db..014544d812 100644
--- a/easybuild/easyconfigs/t/torchvision/torchvision-0.16.0-foss-2023a-CUDA-12.1.1.eb
+++ b/easybuild/easyconfigs/t/torchvision/torchvision-0.21.0-foss-2024a-CUDA-12.6.0.eb
@@ -1,34 +1,57 @@
+easyblock = 'PythonBundle'
+
 name = 'torchvision'
-version = '0.16.0'
-versionsuffix = '-CUDA-%(cudaver)s'
+version = '0.21.0'
+versionsuffix = '-CUDA-12.6.0'
 
 homepage = 'https://github.com/pytorch/vision'
 description = " Datasets, Transforms and Models specific to Computer Vision"
 
-toolchain = {'name': 'foss', 'version': '2023a'}
+toolchain = {'name': 'foss', 'version': '2024a'}
 toolchainopts = {'cstd': 'c++17'}
 
-source_urls = ['https://github.com/pytorch/vision/archive']
-sources = ['v%(version)s.tar.gz']
-patches = [
-    '%(name)s-%(version)s_fix-build-with-FFmpeg6.patch',
-]
-checksums = [
-    {'v0.16.0.tar.gz': '79b30b082237e3ead21e74587cedf4a4d832f977cf7dfeccfb65f67988b12ceb'},
-    {'torchvision-0.16.0_fix-build-with-FFmpeg6.patch':
-     'a49336e7bfa1c950e886852bff37a3ea2146ac7bda87241e3ffb31c5cb869cce'},
-]
-
 builddependencies = [
-    ('CMake', '3.26.3'),
+    ('CMake', '3.29.3'),
 ]
 
 dependencies = [
-    ('CUDA', '12.1.1', '', SYSTEM),
-    ('Python', '3.11.3'),
-    ('FFmpeg', '6.0'),
-    ('Pillow-SIMD', '9.5.0'),
-    ('PyTorch', '2.1.2', versionsuffix),
+    ('Python', '3.12.3'),
+    ('Python-bundle-PyPI', '2024.06'),
+    ('SciPy-bundle', '2024.05'),
+    ('FFmpeg', '7.0.2'),
+    ('Pillow', '10.4.0'),
+    ('PyTorch', '2.6.0', versionsuffix),
+    ('libwebp', '1.4.0'),
+]
+
+exts_list = [
+    ('pytest-mock', '3.14.1', {
+        'source_tmpl': 'pytest_mock-%(version)s.tar.gz',
+        'checksums': ['159e9edac4c451ce77a5cdb9fc5d1100708d2dd4ba3c3df572f14097351af80e'],
+    }),
+    ('torchvision', '0.21.0', {
+        'source_urls': ['https://github.com/pytorch/vision/archive'],
+        'sources': [{'download_filename': 'v%(version)s.tar.gz', 'filename': '%(name)s-%(version)s.tar.gz'}],
+        'patches': [
+            'torchvision-0.17.0_quantized_tol.patch',
+        ],
+        'installopts': '-v',
+        'runtest': (
+            'pytest'
+            ' -m "not xfail"'  # don't run tests that are expected that they might fail
+            ' -k "not test_frame_reading_mem_vs_file"'  # this one hangs
+            '" and not test_antialias_default_warning"'  # flaky test which is removed in later versions
+            '" and not test_videoapi"'  # video support is unstable
+            '" and not avif and not heic"'  # requires torchvision-extra-decoders, libavif PyTorch support, libheic
+            '" and not test_pretrained_pos and not test_equivalent_behavior_weights"'  # tests out of date?
+        ),
+        'testinstall': True,
+        'checksums': [
+            {'torchvision-0.21.0.tar.gz': '0a4a967bbb7f9810f792cd0289a07fb98c8fb5d1303fae8b63e3a6b05d720058'},
+            {'torchvision-0.17.0_quantized_tol.patch':
+             '991b959f7d5e0fd418a163fa3ad828a2b60a64d0ce894cf579847138b13b62af'},
+        ],
+    }),
 ]
 
 moduleclass = 'vis'

[Micket]

I would prefer if we updated Pytorch-bundle instead