diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2020-04-20 14:57:56 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-04-20 14:57:56 -0500 |
| commit | 0db2f006353590374a5107b095623ba8cc9dce5b (patch) | |
| tree | 0e5d6e34ed0e722ef35c425617b22f621ed8aab0 /var | |
| parent | de13ea228407c377b6c42b54f70a17bef8be1854 (diff) | |
| download | spack-0db2f006353590374a5107b095623ba8cc9dce5b.tar.gz spack-0db2f006353590374a5107b095623ba8cc9dce5b.tar.bz2 spack-0db2f006353590374a5107b095623ba8cc9dce5b.tar.xz spack-0db2f006353590374a5107b095623ba8cc9dce5b.zip | |
PyTorch: remove CUDA conflicts (#16136)
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 18 |
1 files changed, 0 insertions, 18 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 31bfb71657..619a621c2b 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -6,7 +6,6 @@ from spack import * -# TODO: try switching to CMakePackage for more control over build class PyTorch(PythonPackage, CudaPackage): """Tensors and Dynamic neural networks in Python with strong GPU acceleration.""" @@ -107,26 +106,9 @@ class PyTorch(PythonPackage, CudaPackage): # see https://github.com/pytorch/pytorch/issues/35149 conflicts('+fbgemm', when='@1.4.0') - cuda_arch_conflict = ('This version of Torch/Caffe2 only supports compute ' - 'capabilities ') - conflicts('cuda_arch=none', when='+cuda', msg='Must specify CUDA compute capabilities of your GPU, see ' 'https://developer.nvidia.com/cuda-gpus') - conflicts('cuda_arch=52', when='@1.3.0:+cuda', - msg=cuda_arch_conflict + '>=5.3') - conflicts('cuda_arch=50', when='@1.3.0:+cuda', - msg=cuda_arch_conflict + '>=5.3') - conflicts('cuda_arch=35', when='@1.3.0:+cuda', - msg=cuda_arch_conflict + '>=5.3') - conflicts('cuda_arch=32', when='@1.3.0:+cuda', - msg=cuda_arch_conflict + '>=5.3') - conflicts('cuda_arch=30', when='@1.3.0:+cuda', - msg=cuda_arch_conflict + '>=5.3') - conflicts('cuda_arch=30', when='@1.2.0:+cuda', - msg=cuda_arch_conflict + '>=3.2') - conflicts('cuda_arch=20', when='@1.0.0:+cuda', - msg=cuda_arch_conflict + '>=3.0') # Required dependencies depends_on('cmake@3.5:', type='build') |