问题描述
运行Github下载的YOLOv5代码中的train.py文件的时候,出现OSError: [WinError 1455] 页面文件太小,无法完成操作。Error loading "D:\Anaconda3\envs\unet\lib\site-packages\torch\lib\cudnn_cnn_infer64_8.dll" or one of its dependencies.
原因分析:
程序中设置的num_workers太大了。一般设置小于电脑cpu的核心数。
在程序中ctrl+f,查找num_workers。最后在程序中找到设置num_workers的语句
parser.add_argument('--workers', type=int, default=8, help='max dataloader workers (per RANK in DDP mode)')
实际的num_workers=workers*2
解决方案:
改小num_workers
parser.add_argument('--workers', type=int, default=2, help='max dataloader workers (per RANK in DDP mode)')
对于我的电脑cpu是8核,设置4都会报错,所以我设置为了2,可以成功运行。