zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add latest version of py-torchvision (#13289)

Browse Source
This commit is contained in:
Adam J. Stewart
2019-10-20 11:12:43 -05:00
committed by GitHub
parent 5995c3f85b
commit de6d53b626
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
var/spack/repos/builtin/packages/py-torchvision/package.py
View File

@@ -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')