diff options
| author | Robert Blake <blake14@llnl.gov> | 2020-09-02 17:52:00 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-09-02 19:52:00 -0500 |
| commit | 638f44d842d14d776fc78633f5c32e4dab808264 (patch) | |
| tree | c580dd4062a63788c7ba0f7a66fb86f6f0f8ca08 | |
| parent | 6d87cbdb524c0fad2100f8fa15b0c76dce432750 (diff) | |
| download | spack-638f44d842d14d776fc78633f5c32e4dab808264.tar.gz spack-638f44d842d14d776fc78633f5c32e4dab808264.tar.bz2 spack-638f44d842d14d776fc78633f5c32e4dab808264.tar.xz spack-638f44d842d14d776fc78633f5c32e4dab808264.zip | |
CUDA_HOME needs to be set, or CUDA is found based on user's PATH. (#18475)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torchvision/package.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-torchvision/package.py b/var/spack/repos/builtin/packages/py-torchvision/package.py index ae6402d1c1..b70e2cd367 100644 --- a/var/spack/repos/builtin/packages/py-torchvision/package.py +++ b/var/spack/repos/builtin/packages/py-torchvision/package.py @@ -86,5 +86,6 @@ class PyTorchvision(PythonPackage): 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) |