MichiGAN icon indicating copy to clipboard operation
MichiGAN copied to clipboard

unable to run it on CPU

Open durgeshmishra2015 opened this issue 2 years ago • 0 comments

I am unable to run it on CPU. It gives following error when running demo.py.

./checkpoints\MichiGAN\SInpaintingModel_gen.pth THCudaCheck FAIL file=..\aten\src\THC\THCGeneral.cpp line=51 error=38 : no CUDA-capable device is detected Traceback (most recent call last): File "demo.py", line 329, in edit orient_stroke = cal_stroke_orient.stroke_to_orient(mask_stroke) File "C:\Users\dmishra\MichiGAN\ui_util\cal_orient_stroke.py", line 143, in stroke_to_orient stroke_mask_tensor = torch.unsqueeze(stroke_mask_tensor, 0).cuda() File "C:\Users\dmishra\Anaconda3\envs\michigan\lib\site-packages\torch\cuda_init_.py", line 162, in _lazy_init torch._C._cuda_init() RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at ..\aten\src\THC\THCGeneral.cpp:51

durgeshmishra2015 avatar Dec 29 '22 07:12 durgeshmishra2015