FaPN
FaPN copied to clipboard
RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:309
When putting the DCNv2 provided by FaPN on my own project, I met the problem of "RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:309". I used the cuda 11.0 and pytorch 1.7, and run on RTX4090, can you help me find the solutions?