diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torchvision/package.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/py-torchvision/package.py b/var/spack/repos/builtin/packages/py-torchvision/package.py index c322ac8f55..d369cef54b 100644 --- a/var/spack/repos/builtin/packages/py-torchvision/package.py +++ b/var/spack/repos/builtin/packages/py-torchvision/package.py @@ -11,7 +11,7 @@ class PyTorchvision(PythonPackage): architectures, and common image transformations for computer vision.""" homepage = "https://github.com/pytorch/vision" - url = "https://github.com/pytorch/vision/archive/v0.3.0.tar.gz" + url = "https://github.com/pytorch/vision/archive/v0.4.0.tar.gz" maintainers = ['adamjstewart'] import_modules = [ @@ -21,13 +21,16 @@ class PyTorchvision(PythonPackage): 'torchvision.models.detection' ] + version('0.4.0', sha256='c270d74e568bad4559fed4544f6dd1e22e2eb1c60b088e04a5bd5787c4150589') version('0.3.0', sha256='c205f0618c268c6ed2f8abb869ef6eb83e5339c1336c243ad321a2f2a85195f0') depends_on('py-setuptools', type='build') depends_on('py-numpy', type=('build', 'run')) depends_on('py-six', type=('build', 'run')) + depends_on('py-torch@1.2.0:', when='@0.4.0:', type=('build', 'run')) depends_on('py-torch@1.1.0:', type=('build', 'run')) depends_on('py-pillow@4.1.1:', type=('build', 'run')) # or py-pillow-simd # Many of the datasets require additional dependencies to use. # These can be installed after the fact. + depends_on('py-scipy', type='test') |