diff --git a/var/spack/repos/builtin/packages/py-torchvision/package.py b/var/spack/repos/builtin/packages/py-torchvision/package.py
index c3589314534..aaca507ba97 100644
--- a/var/spack/repos/builtin/packages/py-torchvision/package.py
+++ b/var/spack/repos/builtin/packages/py-torchvision/package.py
@@ -98,5 +98,10 @@ 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)
+            pytorch_cuda_arch = ';'.join(
+                '{0:.1f}'.format(float(i) / 10.0) for i in
+                self.spec['py-torch'].variants['cuda_arch'].value
+            )
+            env.set('TORCH_CUDA_ARCH_LIST', pytorch_cuda_arch)
         else:
             env.set('FORCE_CUDA', 0)