Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
遇到这个报错,先别急着干其他的,先把device改成CPU试一试,可能就把真正的问题报出来了。
文章共66字 · 阅读需要大约1分钟
一键AI生成摘要,助你高效阅读
问答
·
Compile with TORCH_USE_CUDA_DSA
to enable device-side assertions.
遇到这个报错,先别急着干其他的,先把device改成CPU试一试,可能就把真正的问题报出来了。
即
device='cpu'
然后重新运行代码
更多推荐
已为社区贡献2条内容
所有评论(0)