问题描述:

程序卡死在了dataloader(),但是原本是可以跑的,中途中断了一次之后,后来就不行了。


原因分析:

这种情况应该是属于pytorch多线程锁死,大多数情况是numwork数量多于了cpu的核数。


解决方案:

1.查看CPU个数

cat /proc/cpuinfo | grep "physical id" | uniq | wc -l

2.查看CPU核数

cat /proc/cpuinfo | grep "cpu cores" | uniq

在这里插入图片描述

numwork设置为6。

更多推荐