diff --git a/rlpytorch/model_interface.py b/rlpytorch/model_interface.py
index bcedf172..9a28e52f 100644
--- a/rlpytorch/model_interface.py
+++ b/rlpytorch/model_interface.py
@@ -96,7 +96,7 @@ def add_model(self, key, model, copy=False, cuda=False, gpu_id=None, opt=False,
         self.models[key] = model.clone() if copy else model
         if cuda:
             if gpu_id is not None:
-                self.models[key].cuda(device_id=gpu_id)
+                self.models[key].cuda(device=gpu_id)
             else:
                 self.models[key].cuda()
 
diff --git a/rlpytorch/model_loader.py b/rlpytorch/model_loader.py
index ef471c0e..580eddc7 100644
--- a/rlpytorch/model_loader.py
+++ b/rlpytorch/model_loader.py
@@ -92,7 +92,7 @@ def load_model(self, params):
                     sys.exit(1)
 
         if args.gpu is not None and args.gpu >= 0:
-            model.cuda(device_id=args.gpu)
+            model.cuda(device=args.gpu)
 
         return model