cuda_launch_blocking=1.

runtimeerror: cuda error: device-side assert triggered cuda kernel errors might be asynchronously reported at some other api call,so the stacktrace below might be incorrect. for debugging consider passing cuda_launch_blocking=1.

 

소스코드 최상단에 

 

import os
os.environ['CUDA_LAUNCH_BLOCKING'] = "1"