diff options
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py b/var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py index ca91302ed7..a0ac3fe60f 100644 --- a/var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py +++ b/var/spack/repos/builtin/packages/py-torch-nvidia-apex/package.py @@ -25,6 +25,9 @@ class PyTorchNvidiaApex(PythonPackage, CudaPackage): variant("cuda", default=True, description="Build with CUDA") + # https://github.com/NVIDIA/apex/issues/1498 + conflicts("~cuda") + def setup_build_environment(self, env): if "+cuda" in self.spec: env.set("CUDA_HOME", self.spec["cuda"].prefix) @@ -35,7 +38,7 @@ class PyTorchNvidiaApex(PythonPackage, CudaPackage): ) env.set("TORCH_CUDA_ARCH_LIST", torch_cuda_arch) - def install_options(self, spec, prefix): + def global_options(self, spec, prefix): args = [] if spec.satisfies("^py-torch@1.0:"): args.append("--cpp_ext") |