DCFNet_pytorch icon indicating copy to clipboard operation
DCFNet_pytorch copied to clipboard

Published 20 hours ago • verified publisher icon foolwood

cuFFT error!!!

Open maliangzhibi opened this issue 5 years ago • 4 comments

when I run the command for training, the cuFFT error happened. It happened in the line 47 of net.py and it has a tips of "RuntimeError: cuFFT error: CUFFT_ALLOC_FAILED ". I have no idea and I don't know how to solve it? Anyone else has a solution?

maliangzhibi avatar Jun 01 '19 09:06 maliangzhibi

So, have you installed CUDA support? Or just disable GPU pattern of pytorch. hope help you

tigereatsheep avatar Jun 03 '19 01:06 tigereatsheep

So, have you installed CUDA support? Or just disable GPU pattern of pytorch. hope help you

I have the CUDA support. When I use one GPU for running, it's ok, but in the case of multi-GPU, it's wrong. So, I guess there is a bug in the program, but I can't find it.

maliangzhibi avatar Jun 03 '19 12:06 maliangzhibi

https://github.com/pytorch/pytorch/issues/6511

tigereatsheep avatar Jun 05 '19 00:06 tigereatsheep

@maliangzhibi I have the same question. One GPU is OK,but multi-GPU failed. Have you solved the problem?

Hevans123 avatar Jun 15 '19 10:06 Hevans123