diff --git a/network/network_utils.py b/network/network_utils.py
index fca5135..c3659a1 100644
--- a/network/network_utils.py
+++ b/network/network_utils.py
@@ -12,7 +12,7 @@ import numpy as np
 class BaseNet:
     def set_gpu(self, gpu):
         if gpu >= 0 and torch.cuda.is_available():
-            self.device = torch.device('gpu:%d' % (gpu))
+            self.device = torch.device('cuda:%d' % (gpu))
         else:
             self.device = torch.device('cpu')
         self.to(self.device)