解决[w pthreadpool-ag凯发k8国际

解决[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凯发k8国际为你收集整理的解决[w pthreadpool-cpp.cc:90] warning: leaking caffe2 thread-pool after fork. (function pthreadpool）的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得ag凯发k8国际网站内容还不错，欢迎将ag凯发k8国际推荐给好友。