NVIDIA / vid2vid

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Errors when running on CPU without CUDA

davidvfx07 opened this issue · comments

When I set --gpu_id=-1, I start getting errors from the models when they are initialized like

Traceback (most recent call last):
  File "train.py", line 148, in <module>
    train()
  File "train.py", line 28, in train
    models = create_model(opt)
  File "C:\Users\moish\Desktop\vid2vid\models\models.py", line 76, in create_model
    modelG.initialize(opt)
  File "C:\Users\moish\Desktop\vid2vid\models\vid2vid_model_G.py", line 59, in initialize
    self.n_frames_per_gpu = min(self.opt.max_frames_per_gpu, self.opt.n_frames_total // self.n_gpus) # number of frames in each GPU
ZeroDivisionError: integer division or modulo by zero

When then setting --n_gpus_gen 1 it gets through that error but then comes

Traceback (most recent call last):
  File "train.py", line 148, in <module>
    train()
  File "train.py", line 28, in train
    models = create_model(opt)
  File "C:\Users\moish\Desktop\vid2vid\models\models.py", line 78, in create_model
    modelD.initialize(opt)
  File "C:\Users\moish\Desktop\vid2vid\models\vid2vid_model_D.py", line 22, in initialize
    self.gpu_ids = ([opt.gpu_ids[0]] + opt.gpu_ids[gpu_split_id:]) if opt.n_gpus_gen != len(opt.gpu_ids) else opt.gpu_ids
IndexError: list index out of range

Forcefully setting it to -1 fixes it but then gives me

Traceback (most recent call last):
  File "train.py", line 148, in <module>
    train()
  File "train.py", line 28, in train
    models = create_model(opt)
  File "C:\Users\moish\Desktop\vid2vid\models\models.py", line 78, in create_model
    modelD.initialize(opt)
  File "C:\Users\moish\Desktop\vid2vid\models\vid2vid_model_D.py", line 37, in initialize
    opt.num_D, not opt.no_ganFeat, gpu_ids=self.gpu_ids)
  File "C:\Users\moish\Desktop\vid2vid\models\networks.py", line 66, in define_D
    netD.cuda(gpu_ids[0])
  File "C:\Users\moish\.conda\envs\vid2vid\lib\site-packages\torch\nn\modules\module.py", line 304, in cuda
    return self._apply(lambda t: t.cuda(device))
  File "C:\Users\moish\.conda\envs\vid2vid\lib\site-packages\torch\nn\modules\module.py", line 201, in _apply
    module._apply(fn)
  File "C:\Users\moish\.conda\envs\vid2vid\lib\site-packages\torch\nn\modules\module.py", line 201, in _apply
    module._apply(fn)
  File "C:\Users\moish\.conda\envs\vid2vid\lib\site-packages\torch\nn\modules\module.py", line 223, in _apply
    param_applied = fn(param)
  File "C:\Users\moish\.conda\envs\vid2vid\lib\site-packages\torch\nn\modules\module.py", line 304, in <lambda>
    return self._apply(lambda t: t.cuda(device))
RuntimeError: Device index must not be negative

I'm stuck y'all.