init

APDrawingGAN2/options/base_options.py

@@ -178,7 +178,7 @@ class BaseOptions():
             if id >= 0:
                 opt.gpu_ids.append(id)
         if len(opt.gpu_ids) > 0:
-            torch.cuda.set_device(opt.gpu_ids[0])
+            #torch.cuda.set_device(opt.gpu_ids[0])
         
         # set gpu ids
         str_ids = opt.gpu_ids_p.split(',')
@@ -186,7 +186,7 @@ class BaseOptions():
         for str_id in str_ids:
             id = int(str_id)
             if id >= 0:
-                opt.gpu_ids_p.append(id)
+                #opt.gpu_ids_p.append(id)
 
         self.opt = opt
         return self.opt

app.py

@@ -167,8 +167,8 @@ def main():
     opt.netG = 'resnet_9blocks'
     opt.which_epoch = 150
     opt.how_many = 1000
-    opt.gpu_ids = ''
-    opt.gpu_ids_p = ''
+    opt.gpu_ids = '-1'
+    opt.gpu_ids_p = '-1'
     opt.imagefolder = 'images-single'
 
     opt.checkpoints_dir = checkpoint_dir