Update gen_images.py

gen_images.py

@@ -19,6 +19,8 @@ import PIL.Image
 import torch
 
 import legacy
+os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Disables CUDA completely
+
 
 #----------------------------------------------------------------------------
 
@@ -103,7 +105,7 @@ def generate_images(
     """
 
     print('Loading networks from "%s"...' % network_pkl)
-    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    device = torch.device('cpu')
     with dnnlib.util.open_url(network_pkl) as f:
         G = legacy.load_network_pkl(f)['G_ema'].to(device) # type: ignore