diff options
| author | range3 <white.line.oar@gmail.com> | 2022-04-17 12:06:43 +0900 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-04-16 21:06:43 -0600 |
| commit | a405a6efe3f51ba15555bbaf6d631c2c30664d6b (patch) | |
| tree | f39bf877eafe70c34491a592051ff2921b0c6748 | |
| parent | b0c8affbd943b6d86bf650ceab7faa16eacaddcb (diff) | |
| download | spack-a405a6efe3f51ba15555bbaf6d631c2c30664d6b.tar.gz spack-a405a6efe3f51ba15555bbaf6d631c2c30664d6b.tar.bz2 spack-a405a6efe3f51ba15555bbaf6d631c2c30664d6b.tar.xz spack-a405a6efe3f51ba15555bbaf6d631c2c30664d6b.zip | |
py-torch: fix cuda dependencies (#30117)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 6 |
1 files changed, 3 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 59248a3191..8190e01102 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -139,9 +139,9 @@ class PyTorch(PythonPackage, CudaPackage): # Optional dependencies # https://discuss.pytorch.org/t/compiling-1-10-1-from-source-with-gcc-11-and-cuda-11-5/140971 depends_on('cuda@9.2:', when='@1.11:+cuda', type=('build', 'link', 'run')) - depends_on('cuda@9.2:11.4', when='@1.6:+cuda', type=('build', 'link', 'run')) - depends_on('cuda@9:11.4', when='@1.1:+cuda', type=('build', 'link', 'run')) - depends_on('cuda@7.5:11.4', when='+cuda', type=('build', 'link', 'run')) + depends_on('cuda@9.2:11.4', when='@1.6:1.10+cuda', type=('build', 'link', 'run')) + depends_on('cuda@9:11.4', when='@1.1:1.5+cuda', type=('build', 'link', 'run')) + depends_on('cuda@7.5:11.4', when='@:1.0+cuda', type=('build', 'link', 'run')) depends_on('cudnn@6:7', when='@:1.0+cudnn') depends_on('cudnn@7.0:7', when='@1.1:1.5+cudnn') depends_on('cudnn@7:', when='@1.6:+cudnn') |