5,321
edits
PyTorch: Difference between revisions
→Debugging
| Line 114: | Line 114: | ||
==Debugging== | ==Debugging== | ||
{{see also | Debugging ML Models}} | {{see also | Debugging ML Models}} | ||
If you get a cuda kernel error, you can rerun with the environment variable <code>CUDA_LAUNCH_BLOCKING=1</code> to get the correct line. | If you get a cuda kernel error, you can rerun with the environment variable <code>CUDA_LAUNCH_BLOCKING=1</code> to get the correct line in the stack trace. | ||
<pre> | <pre> | ||
CUDA_LAUNCH_BLOCKING=1 python app.py | CUDA_LAUNCH_BLOCKING=1 python app.py | ||