search

hezhangsprinter / DCPDN

Densely Connected Pyramid Dehazing Network (CVPR'2018)
401 stars 112 forks source link

run without cuda and GPU #11

Open mod1998 opened 6 years ago

mod1998 commented 6 years ago

when I run "python demo.py --dataroot ./facades/nat_new4 --valDataroot ./facades/nat_new4 --netG ./demo_model/netG_epoch_8.pth" without cuda and GPU,I get that: self.prev_idx = torch._C._cuda_getDevice() RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at torch/csrc/cuda/Module.cpp:51

hezhangsprinter commented 6 years ago

If you want to run without cuda and GPU, please change the code to non-gpu version. For example, please remove xx.cuda() and make all the tensor and loss functions in cpu version.

ShiqiLi047 commented 4 years ago

Have you get the cpu version of this code?

chinmayee521 commented 4 years ago

Can you please help with the cpu version of this code?

daiqingyue commented 1 year ago

Replace .cuda() with .to(device) or .cpu() in your code