stylegan2-pytorch icon indicating copy to clipboard operation
stylegan2-pytorch copied to clipboard

RuntimeError: no valid convolution algorithms available in CuDNN

Open leo4048111 opened this issue 1 year ago • 4 comments

File "/home/leo/GAN2Shape/gan2shape/stylegan2/stylegan2-pytorch/model.py", line 350, in forward out = self.conv(input, style) File "/home/leo/miniconda3/envs/cv_exp_py3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/leo/GAN2Shape/gan2shape/stylegan2/stylegan2-pytorch/model.py", line 272, in forward out = F.conv_transpose2d(input, weight, padding=0, stride=2, groups=batch) RuntimeError: no valid convolution algorithms available in CuDNN

leo4048111 avatar May 29 '23 13:05 leo4048111

I have the same problem. Do you solve it?

huangzy-55 avatar Aug 19 '23 05:08 huangzy-55

I have the same problem. Do you solve it?

I degraded pytorch + cudnn versions and it seemed to work.

leo4048111 avatar Aug 19 '23 06:08 leo4048111

I have the same problem. Do you solve it?

I degraded pytorch + cudnn versions and it seemed to work.

which version? could you tell me please?

huangzy-55 avatar Aug 19 '23 06:08 huangzy-55

I have the same problem. Do you solve it?

I degraded pytorch + cudnn versions and it seemed to work.

which version? could you tell me please?

PyTorch 1.3.1 + cuda 10.1 + cudnn 8, just as the readme stated. Also consider using legacy GPUs such as gtx1080.

leo4048111 avatar Aug 19 '23 08:08 leo4048111