search

foolwood / DCFNet_pytorch

DCFNet: Discriminant Correlation Filters Network for Visual Tracking
https://arxiv.org/pdf/1704.04057.pdf
MIT License
207 stars 60 forks source link

cuFFT error!!! #16

Open maliangzhibi opened 5 years ago

maliangzhibi commented 5 years ago

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?

tigereatsheep commented 5 years ago

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

maliangzhibi commented 5 years ago

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.

tigereatsheep commented 5 years ago

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

Hevans123 commented 5 years ago

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