diff --git a/var/spack/repos/builtin/packages/py-torchvision/package.py b/var/spack/repos/builtin/packages/py-torchvision/package.py
index ae6402d1c1d..b70e2cd3678 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 @@ 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)