2024年5月15日
pytorch调试时CUDA报错，如何精确定位
摘要： 由于pytorch的python语句执行和CUDA后端实际上是异步执行的，所以当CUDA runtime报错时，并不一定能正确地反映在python抛出的错误语句中，因此，可以添加一个环境变量来实现： CUDA_LAUNCH_BLOCKING=1 这里再补充一些关于cuda和pytorch异步执行的知 阅读全文
posted @ 2024-05-15 20:15 王冰冰 阅读(52) 评论(3) 推荐(0) 编辑