zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-torch: CUDA 9.2+ required for 1.6+ (#24808)

Browse Source
This commit is contained in:
Adam J. Stewart 2021-07-13 12:46:07 -05:00 committed by GitHub
parent 657b3ec052
commit 9828df7335
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
var/spack/repos/builtin/packages/py-torch/package.py
View File

@ -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')