summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/py-torchvision/package.py5
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')