pytorch RuntimeError: unable to open shared memory object in read-write mode: Too many open files

在使用PyTorch的多进程数据加载时,由于自定义的collate_fn函数将list内的list转换为Tensor,导致RuntimeError:unabletoopensharedmemoryobjectinread-writemode:Toomanyopenfiles。问题的根本原因在于转换的Tensor过多。通过设置torch.multiprocessing的sharing_strategy为file_system可以缓解此问题,但最终解决方法是优化collate_fn函数,避免过多的Tensor操作。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

RuntimeError: unable to open shared memory object in read-write mode: Too many open files (24)

网上的解决办法是通过添加下面两行解决

import torch.multiprocessing

torch.multiprocessing.set_sharing_strategy('file_system')

后来发现实际上是我们自定义了dataloader传入的collate_fn处理有问题,如果里面有把list里面又有list,我把list里面的那个list转成Tensor就会导致这个问题,原因是list里面的tensor太多了,参考解决地址torch.tensors in torch.multiprocessing · Issue #11899 · pytorch/pytorch · GitHub

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值