「Bug」“OSError [WinError 1455] 页面文件太小,无法完成操作” 解决方案

训练YOLOv5模型时出现错误

OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading "F:\anaconda\envs\yolov5\lib\site-packages\torch\lib\shm.dll" or one of its dependencies.

解决方法

修改train.py中配置项workers的默认值,从初始值8修改为0(也可以试试1或2等较小的数)。

parser.add_argument('--workers', type=int, default=8, help='max dataloader workers (per RANK in DDP mode)')

错误原因

workers_per_gpu=2, # 每个gpu分配的线程数,给每一个GPU喂数据的进程,GPU性能越强,取值越大,这样才能充分利用GPU的算力。如果用自己的电脑训练,这个值需要取小一些。

猜你喜欢

转载自blog.csdn.net/ViatorSun/article/details/129264527
今日推荐