Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.
遇到这个报错,先别急着干其他的,先把device改成CPU试一试,可能就把真正的问题报出来了。


device='cpu'
然后重新运行代码

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐