THCBlas.cu:450

Asked Jan 15 '21 at 04:27

Active Aug 23 '21 at 14:23

Viewed 1,799 times

My program works properly with two GPUs but raise this error when reached to torch.bmm

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 could I solve this issue ?

edited Jan 15 '21 at 05:46

talonmies

70,661
34
192
269

asked Jan 15 '21 at 04:27

Roqýah Ȝbđeen

1

check this link https://github.com/pytorch/pytorch/issues/20860 – Prabhat Kumar Sahu Jan 15 '21 at 06:46

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

0 Answers0