diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2019-12-11 20:16:52 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-12-11 20:16:52 -0600 |
| commit | 1f9d37c3464cf9867ed8e7960d27d11eff0d323a (patch) | |
| tree | 755599084859492aa5bca823881cd9a5eacb090c | |
| parent | 7ec1917d9b7141b5eda59a5fbd0890936150653a (diff) | |
| download | spack-1f9d37c3464cf9867ed8e7960d27d11eff0d323a.tar.gz spack-1f9d37c3464cf9867ed8e7960d27d11eff0d323a.tar.bz2 spack-1f9d37c3464cf9867ed8e7960d27d11eff0d323a.tar.xz spack-1f9d37c3464cf9867ed8e7960d27d11eff0d323a.zip | |
TensorFlow: must specify CUDA compute capabilities (#14118)
* TensorFlow: must specify CUDA compute capabilities
* Correctly restrict cuda_arch version ranges
| -rw-r--r-- | var/spack/repos/builtin/packages/py-tensorflow/package.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/var/spack/repos/builtin/packages/py-tensorflow/package.py b/var/spack/repos/builtin/packages/py-tensorflow/package.py index f139cf6cb3..69dabac970 100644 --- a/var/spack/repos/builtin/packages/py-tensorflow/package.py +++ b/var/spack/repos/builtin/packages/py-tensorflow/package.py @@ -205,6 +205,11 @@ class PyTensorflow(Package, CudaPackage): conflicts('+computecpp', when='~opencl') conflicts('+rocm', when='@:1.11') conflicts('+cuda', when='platform=darwin', msg='There is no GPU support for macOS') + 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=20', when='@1.12.1,1.14:', msg='TensorFlow only supports compute capabilities >= 3.5') + conflicts('cuda_arch=30', when='@1.12.1,1.14:', msg='TensorFlow only supports compute capabilities >= 3.5') + conflicts('cuda_arch=32', when='@1.12.1,1.14:', msg='TensorFlow only supports compute capabilities >= 3.5') + conflicts('cuda_arch=20', when='@1.4:1.12.0,1.12.2:1.12.3', msg='Only compute capabilities 3.0 or higher are supported') conflicts('+tensorrt', when='@:1.5') conflicts('+tensorrt', when='~cuda') conflicts('+tensorrt', when='platform=darwin', msg='Currently TensorRT is only supported on Linux platform') @@ -423,10 +428,9 @@ class PyTensorflow(Package, CudaPackage): # Please note that each additional compute capability significantly # increases your build time and binary size, and that TensorFlow # only supports compute capabilities >= 3.5 - if spec.variants['cuda_arch'].value != 'none': - capabilities = ','.join('{0:.1f}'.format( - float(i) / 10.0) for i in spec.variants['cuda_arch'].value) - env.set('TF_CUDA_COMPUTE_CAPABILITIES', capabilities) + capabilities = ','.join('{0:.1f}'.format( + float(i) / 10.0) for i in spec.variants['cuda_arch'].value) + env.set('TF_CUDA_COMPUTE_CAPABILITIES', capabilities) else: env.set('TF_NEED_CUDA', '0') |