diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2021-03-05 03:21:39 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-03-05 10:21:39 +0100 |
| commit | 5fb0ff3906ff949d689493d8b218d87af8e69476 (patch) | |
| tree | 34d9d5621e8536bfb738bc87c7d338aed8af2f6e | |
| parent | 4261de543407a22ab6284956c15b6b015e528c93 (diff) | |
| download | spack-5fb0ff3906ff949d689493d8b218d87af8e69476.tar.gz spack-5fb0ff3906ff949d689493d8b218d87af8e69476.tar.bz2 spack-5fb0ff3906ff949d689493d8b218d87af8e69476.tar.xz spack-5fb0ff3906ff949d689493d8b218d87af8e69476.zip | |
py-torch: add v1.8.0 (#22101)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index afc89b6f8c..9b1ee499e1 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -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')) |