diff --git a/GDQN/utils.py b/GDQN/utils.py
index f4e7950..338d2d1 100644
--- a/GDQN/utils.py
+++ b/GDQN/utils.py
@@ -1,3 +1,4 @@
+from gpuinfo.nvidia import get_gpus
 import psutil