diff --git a/cscs-checks/prgenv/cuda/cuda_aware_mpi.py b/cscs-checks/prgenv/cuda/cuda_aware_mpi.py
index 5a0169d4f1..d992f73bae 100644
--- a/cscs-checks/prgenv/cuda/cuda_aware_mpi.py
+++ b/cscs-checks/prgenv/cuda/cuda_aware_mpi.py
@@ -47,6 +47,10 @@ def set_valid_prog_environs(self):
     def set_compilers(self):
         if self.current_environ.name == 'PrgEnv-pgi':
             self.build_system.cflags = ['-std=c99', ' -O3']
+        elif self.current_environ.name == 'PrgEnv-nvidia':
+            self.variables = {
+                'CUDA_HOME': '$CRAY_NVIDIA_PREFIX/cuda'
+            }
 
         gcd_flgs = (
             '-gencode arch=compute_{0},code=sm_{0}'.format(self.gpu_arch)