cublas runtime error: the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:334 #54
Description
Excuse me, when I run demo.py, I get the error "cublas runtime error: the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:334", how should I solve this? Is it because of a problem with my environment? My environment uses cuda110, pytorch1.7.0-cu110-py37, python=3.7.6. Thanks a lot, please advise.