summaryrefslogtreecommitdiff
path: root/var
diff options
context:
space:
mode:
Diffstat (limited to 'var')
-rw-r--r--var/spack/repos/builtin/packages/py-torch/package.py8
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'))