有时候写代码,尤其是深度学习使用gpu的代码,报错很反人类,十几种类型的track有可能吐出来的报错信息都是一样的,而且大多很抽象。。。
两步让程序报错说人话:
程序开头加入:
viron['CUDA_LAUNCH_BLOCKING'] = '1'
把.to('cuda')
变成.to('cpu')
就能获得正常的track
本文发布于:2024-01-31 00:07:48,感谢您对本站的认可!
本文链接:https://www.4u4v.net/it/170663086923815.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
留言与评论(共有 0 条评论) |