正确解决pytorch报错:RuntimeError: DataLoader worker (pid(s) ***, ***, ***, ***) exited unexpectedly,亲测有效

错误原因:torch.utils.data.DataLoader中的num_workers错误
将num_workers改为0即可,0是默认值。
num_workers是用来指定开多进程的数量,默认值为0,表示不启用多进程。

若:
将num_workers设置为0,程序报错,并提示设置环境变量KMP_DUPLICATE_LIB_OK=TRUE
那你可以在设置环境变量KMP_DUPLICATE_LIB_OK=TRUE
或者使用临时环境变量:(在代码开始处添加这行代码)

os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'

你可能感兴趣的:(pytorch)