diff --git a/mpi_info/Makefile b/mpi_info/Makefile
index 6f776f48f41fb049dbad26bf91b10a75458b7e86..2cd102b199f08c79521a1efd6e7b9dc8c79dbc67 100644
--- a/mpi_info/Makefile
+++ b/mpi_info/Makefile
@@ -1,2 +1,2 @@
 mpi_info: mpi_info.c
-	mpicc -o $@ $<
+	mpicc $< -lcudart -o $@
diff --git a/mpi_info/load_modules.sh b/mpi_info/load_modules.sh
index 81909335de558a56da2a58a5fd9d86f03d458ce0..cbba5b3e1b28ba34dc9a5e28751267ac33a6421b 100755
--- a/mpi_info/load_modules.sh
+++ b/mpi_info/load_modules.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
 
 module purge
-module load spack git gcc/7.3.0 openmpi/3.1.4-pmi-cuda-ucx
+module load spack git gcc/7.3.0 cuda/10.1 openmpi/3.1.4-pmi-cuda-ucx
 
diff --git a/mpi_info/mpi_info.c b/mpi_info/mpi_info.c
index 1ec46deb705616595d894788826c4381a6e1e0cf..faabad9c5142f76b549a16f5373ae68cdc9c09d6 100644
--- a/mpi_info/mpi_info.c
+++ b/mpi_info/mpi_info.c
@@ -1,6 +1,7 @@
 #include "mpi.h"
 #include <stdio.h>
 #include <stdlib.h>
+#include <cuda_runtime.h>
 
 int main(int argc, char *argv[]) {
     char* local_rank_str = NULL;