diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2021-07-13 12:46:07 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-07-13 10:46:07 -0700 |
| commit | 9828df73354b9b41fb38ae3d90d39f4da6dea4ea (patch) | |
| tree | 3e91f68c783edf05e11f99afb364090872e49ee8 /var | |
| parent | 657b3ec0525eef7ba0ef99f58c9b481b3979abf8 (diff) | |
| download | spack-9828df73354b9b41fb38ae3d90d39f4da6dea4ea.tar.gz spack-9828df73354b9b41fb38ae3d90d39f4da6dea4ea.tar.bz2 spack-9828df73354b9b41fb38ae3d90d39f4da6dea4ea.tar.xz spack-9828df73354b9b41fb38ae3d90d39f4da6dea4ea.zip | |
py-torch: CUDA 9.2+ required for 1.6+ (#24808)
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 1 |
1 files changed, 1 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 63c7cac7cc..1a7b414af5 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -150,6 +150,7 @@ class PyTorch(PythonPackage, CudaPackage): # Optional dependencies depends_on('cuda@7.5:', when='+cuda', type=('build', 'link', 'run')) depends_on('cuda@9:', when='@1.1:+cuda', type=('build', 'link', 'run')) + depends_on('cuda@9.2:', when='@1.6:+cuda', type=('build', 'link', 'run')) depends_on('cudnn@6.0:7.999', when='@:1.0.999+cudnn') depends_on('cudnn@7.0:7.999', when='@1.1.0:1.5.999+cudnn') depends_on('cudnn@7.0:', when='@1.6.0:+cudnn') |