ReDet icon indicating copy to clipboard operation
ReDet copied to clipboard

Published 20 hours ago • verified publisher icon csuhan

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

Open Sunnyhhhhhh opened this issue 3 years ago • 2 comments

When testing the dataset using the code provided in the GETTING_STARTED.md documentation, I encountered the following problems: RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:450

Sunnyhhhhhh avatar May 09 '22 03:05 Sunnyhhhhhh

I guess you need to change the version of torch. We use torch==1.3 by default.

csuhan avatar May 09 '22 07:05 csuhan

Thank you very much. I will make modifications according to your suggestions.

Sunnyhhhhhh avatar May 09 '22 08:05 Sunnyhhhhhh