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

py-torch: add v1.8.0 (#22101)

Browse Source
This commit is contained in:
Adam J. Stewart 2021-03-05 03:21:39 -06:00 committed by GitHub
parent 4261de5434
commit 5fb0ff3906
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
var/spack/repos/builtin/packages/py-torch/package.py
View File

@ -20,6 +20,7 @@ class PyTorch(PythonPackage, CudaPackage):
import_modules = ['torch', 'torch.autograd', 'torch.nn', 'torch.utils']
version('master', branch='master', submodules=True)
version('1.8.0', tag='v1.8.0', submodules=True)
version('1.7.1', tag='v1.7.1', submodules=True)
version('1.7.0', tag='v1.7.0', submodules=True)
version('1.6.0', tag='v1.6.0', submodules=True)
@ -93,9 +94,10 @@ class PyTorch(PythonPackage, CudaPackage):
# Use Ninja generator to speed up build times
# Automatically used if found
depends_on('ninja@1.5:', type='build')
depends_on('python@3.6.1:', when='@1.6:', type=('build', 'run'))
depends_on('python@3.5:', when='@1.5:', type=('build', 'run'))
depends_on('python@2.7:2.8,3.5:', type=('build', 'run'))
depends_on('python@3.6.2:', when='@1.7.1:', type=('build', 'link', 'run'))
depends_on('python@3.6.1:', when='@1.6:', type=('build', 'link', 'run'))
depends_on('python@3.5:', when='@1.5:', type=('build', 'link', 'run'))
depends_on('python@2.7:2.8,3.5:', type=('build', 'link', 'run'))
depends_on('py-setuptools', type=('build', 'run'))
depends_on('py-numpy', type=('build', 'run'))
depends_on('py-future', when='@1.5:', type=('build', 'run'))