解决[w pthreadpool-凯发ag旗舰厅登录网址下载

解决[w pthreadpool-cpp.cc:90] warning: leaking caffe2 thread-pool after fork. (function pthreadpool）

pytorch 1.9遇到如下问题

问题：

[w pthreadpool-cpp.cc:90] warning: leaking caffe2 thread-pool after fork. (function pthreadpool)
[w pthreadpool-cpp.cc:90] warning: leaking caffe2 thread-pool after fork. (function pthreadpool)
[w pthreadpool-cpp.cc:90] warning: leaking caffe2 thread-pool after fork. (function pthreadpool)
[w pthreadpool-cpp.cc:90] warning: leaking caffe2 thread-pool after fork. (function pthreadpool)

解决方法：
dataloader函数中参数改成pin_memory = false

改前：

dataloader(val_dataset, shuffle=true, batch_size=batch_size, num_workers=4, pin_memory=true,drop_last=true, collate_fn=yolo_dataset_collate)

改后：

dataloader(val_dataset, shuffle=true, batch_size=batch_size, num_workers=4, pin_memory=false,drop_last=true, collate_fn=yolo_dataset_collate)

参考
https://blog.csdn.net/forest_world/article/details/118977692

总结

以上是凯发ag旗舰厅登录网址下载为你收集整理的解决[w pthreadpool-cpp.cc:90] warning: leaking caffe2 thread-pool after fork. (function pthreadpool）的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得凯发ag旗舰厅登录网址下载网站内容还不错，欢迎将凯发ag旗舰厅登录网址下载推荐给好友。