在使用多进程训练时,报错RuntimeError: Attempted to send CUDA tensor received from another process
原因是 dataloader/dataset.py 构建这块出的问题,因为dataset中构建batch时传入的一个tensor被加载到cuda上面了(不应该放在cuda上),导致后续进行多进程处理时出现了 GPU上的进程共享问题
RuntimeError: Attempted to send CUDA tensor received from another process
最新推荐文章于 2024-07-12 23:11:20 发布