【BUG调试】——OSError: Caught OSError in DataLoader worker process 0
目錄
問(wèn)題描述:
參考鏈接
問(wèn)題分析
解決方案
出現(xiàn)情況
問(wèn)題描述:
在使用pytorch搭建了VGG從頭開(kāi)始訓(xùn)練時(shí)出現(xiàn)了以下問(wèn)題:
OSError: Caught OSError in DataLoader worker process 0
參考鏈接
參考up主視頻:4.2 使用pytorch搭建VGG網(wǎng)絡(luò)_嗶哩嗶哩_bilibili
參考博客文章:解決 OSError: cannot identify image file 或者 Caught TypeError in DataLoader worker process 8_夢(mèng)墜凡塵-CSDN博客跑模型在讀取數(shù)據(jù)是出錯(cuò)如題,找了兩個(gè)多小時(shí)干到深夜依然不能解!!!本地把數(shù)據(jù)過(guò)了一遍都是OK的,代碼也是一樣!!!一模一樣的代碼一模一樣的數(shù)據(jù)為何本地OK,服務(wù)器就掛了????最后網(wǎng)上給百度說(shuō)是可能是Pillow版本的原因本地是Pillow == 5.2.0服務(wù)器是5.1.0最后將服務(wù)器升級(jí)為 Pillow5.2.0==真的解決了!!!!!!!!!謎之答案!!!!!浪費(fèi)晚上兩個(gè)多小時(shí)。心痛!謹(jǐn)記!...https://blog.csdn.net/c2250645962/article/details/108658892
問(wèn)題分析
參考上述資源后,發(fā)現(xiàn)在Windows系統(tǒng)下線(xiàn)程只能設(shè)置為0,在Linux或者其他系統(tǒng)上可以設(shè)置為非0值
解決方案
修改前:
# 加載訓(xùn)練數(shù)據(jù)集train_loader = torch.utils.data.DataLoader(train_dataset,batch_size=batch_size, shuffle=True,num_workers=nw)修改后:
train_loader = torch.utils.data.DataLoader(train_dataset,batch_size=batch_size, shuffle=True,num_workers=0)最后問(wèn)題解決,可以正常運(yùn)行
出現(xiàn)情況
由于線(xiàn)程設(shè)置為0,所以相對(duì)于其他系統(tǒng)多線(xiàn)程訓(xùn)練的速度會(huì)比較慢
創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎(jiǎng)勵(lì)來(lái)咯,堅(jiān)持創(chuàng)作打卡瓜分現(xiàn)金大獎(jiǎng)總結(jié)
以上是生活随笔為你收集整理的【BUG调试】——OSError: Caught OSError in DataLoader worker process 0的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: [HAOI2016]食物链
- 下一篇: 数据中心建筑设计及功能区布局的注意事项