summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/py-torch/package.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py
index 619a621c2b..6428d6cb8e 100644
--- a/var/spack/repos/builtin/packages/py-torch/package.py
+++ b/var/spack/repos/builtin/packages/py-torch/package.py
@@ -50,6 +50,7 @@ class PyTorch(PythonPackage, CudaPackage):
]
version('master', branch='master', submodules=True)
+ version('1.5.0', tag='v1.5.0', submodules=True)
version('1.4.1', tag='v1.4.1', submodules=True)
# see https://github.com/pytorch/pytorch/issues/35149
version('1.4.0', tag='v1.4.0', submodules=True,
@@ -115,6 +116,7 @@ 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.5:', when='@1.5:', type=('build', 'run'))
depends_on('python@2.7:2.8,3.5:', type=('build', 'run'))
depends_on('py-setuptools', type='build')
depends_on('py-numpy', type=('build', 'run'))