Skip to content
Snippets Groups Projects
Commit 99090875 authored by Alexandre Strube's avatar Alexandre Strube
Browse files

torchvision needs to be 0.8.2

parent 76b2772f
Branches
No related tags found
No related merge requests found
easyblock = 'PythonPackage'
name = 'torchvision'
version = '0.8.1'
versionsuffix = '-Python-%(pyver)s'
homepage = 'http://pytorch.org/'
description = """The torchvision package consists of popular datasets, model architectures, and common image
transformations for computer vision."""
site_contacts = 'a.strube@fz-juelich.de'
toolchain = {'name': 'gcccoremkl', 'version': '9.3.0-2020.2.254'}
download_dep_fail = True
source_urls = [
'https://github.com/pytorch/vision/archive',
]
sources = [
'v%(version)s.tar.gz'
]
dependencies = [
('Python', '3.8.5'),
('SciPy-Stack', '2020', versionsuffix),
('PyTorch', '1.7.0', versionsuffix),
('Pillow-SIMD', '7.0.0.post3', versionsuffix),
]
options = {'modulename': 'torchvision'}
sanity_check_paths = {
'files': [],
'dirs': ['lib/python%(pyshortver)s/site-packages'],
}
# Add a property
modluafooter = """
add_property("arch","gpu")
"""
moduleclass = 'devel'
......@@ -100,3 +100,6 @@ software:
- name: 'JAX'
owner: 'strube1'
base: True
- name: 'torchvision'
owner: 'strube1'
base: 'true'
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment