search

KupynOrest / DeblurGAN

Image Deblurring using Generative Adversarial Networks
Other
2.51k stars 516 forks source link

RuntimeError: CUDNN_STATUS_INTERNAL_ERROR #53

Open Tz-george opened 6 years ago

Tz-george commented 6 years ago

I input the following command to train my data, but have some error. CUDA_VISIBLE_DEVICE=3 python train.py --dataroot blurred_sharp/ --learn_residual --resize_or_crop crop --fineSize 265 --no_html

error: Traceback (most recent call last): File "train.py", line 59, in train(opt, data_loader, model, visualizer) File "train.py", line 22, in train model.optimize_parameters() File "/home/liaochunxiu/DeblurGAN-master/models/conditional_gan_model.py", line 104, in optimize_parameters self.forward() File "/home/liaochunxiu/DeblurGAN-master/models/conditional_gan_model.py", line 76, in forward self.fake_B = self.netG.forward(self.real_A) File "/home/liaochunxiu/DeblurGAN-master/models/networks.py", line 146, in forward output = self.model(input) File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call result = self.forward(*input, *kwargs) File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/container.py", line 67, in forward input = module(input) File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call result = self.forward(input, **kwargs) File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 282, in forward self.padding, self.dilation, self.groups) File "/usr/local/lib/python3.6/site-packages/torch/nn/functional.py", line 90, in conv2d return f(input, weight, bias) RuntimeError: CUDNN_STATUS_INTERNAL_ERROR

Can you tell me how to do for this? Thanks~

Gasoonjia commented 6 years ago

import torch以后手动加入torch.backends.cudnn.enabled = False应该能够解决。 上面那个方法应该是治标不治本。 感觉是pytorch没抓到CUDNN或者CUDNN版本不对 试试看手动添加CUDNN？这个才是治本的方法 都是中国人就不扯英文了hhh