提示是这样的:CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1
Compile with TORCH_USE_CUDA_DAS to enable device-side assertions