MichiGAN icon indicating copy to clipboard operation
MichiGAN copied to clipboard

Published 20 hours ago • verified publisher icon tzt101

problem with running deomo

Open jakeyahn opened this issue 3 years ago • 1 comments

Traceback (most recent call last): File "demo.py", line 521, in opt = DemoOptions().parse() File "/home/student/michigan_test/MichiGAN/options/base_options.py", line 235, in parse torch.cuda.set_device(opt.gpu_ids[0]) File "/home/student/anaconda3/envs/michigan_env/lib/python3.7/site-packages/torch/cuda/init.py", line 311, in set_device torch._C._cuda_setDevice(device) RuntimeError: CUDA error: invalid device ordinal

I tried cuda_launch_blocking=1 but still doesnt work..

jakeyahn avatar Nov 26 '21 12:11 jakeyahn

In baseoptions line 235 try changing the line to torch.cuda.set_device(int('0'))

SparshDrolia22 avatar Nov 16 '22 13:11 SparshDrolia22