RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:450

heykeetae / Self-Attention-GAN

Pytorch implementation of Self-Attention Generative Adversarial Networks (SAGAN)

2.51k stars 470 forks source link

RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:450 #55

Open RokiaAbdeen opened 3 years ago

RokiaAbdeen commented 3 years ago

I got this error while trying to run code on multiple gpu energy = torch.bmm(proj_query,proj_key) # transpose check RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:450 how to solve this issue?