hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

CUDA_HOME needs to be set, or CUDA is found based on user's PATH. (#18475)

Browse source
This commit is contained in:
Robert Blake 2020-09-02 17:52:00 -07:00 committed by GitHub
parent 6d87cbdb52
commit 638f44d842
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-torchvision/package.py
View file

@ -86,5 +86,6 @@ def setup_build_environment(self, env):
if '+cuda' in self.spec['py-torch']:
env.set('FORCE_CUDA', 1)
env.set('CUDA_HOME', self.spec['cuda'].prefix)
else:
env.set('FORCE_CUDA', 0)