- 博客(5)
- 收藏
- 关注
原创 CUDA error: device-side assert triggered
用nn.parallel.DistributedDataParallel 跑Diffusion模型的时候报错如图,感觉以前也报过类似的错,困扰了一天不知道原因是什么,感觉是跟DDP有关。看了一下别人的Blog,不使用GPU,“CUDA_VISIBLE_DEVICES”:“”,可以让CUDA抛出详细错误,但是我的模型是DDP形式的,没办法直接使用CPU,要改的地方比较多。这是因为CUDA_LAUNCH_BLOCKING是CUDA的环境变量,设为1可以disable CUDA执行的异步性。
2023-06-13 21:11:11
287
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人