欢迎访问 生活随笔!

ag凯发k8国际

当前位置: ag凯发k8国际 > 人工智能 > caffe >内容正文

caffe

warning: leaking caffe2 thread-ag凯发k8国际

发布时间:2024/3/7 caffe 59 豆豆
ag凯发k8国际 收集整理的这篇文章主要介绍了 warning: leaking caffe2 thread-pool after fork 小编觉得挺不错的,现在分享给大家,帮大家做个参考.

warning: leaking caffe2 thread-pool after fork

详细可以参考这里:https://discuss.pytorch.org/t/dataloader-when-num-worker-0-there-is-bug/25643/37

我的情况是在pytorch 1.9.1,cuda 11.1, rtx 3090 显卡上出现这个问题。
解决方法: 将 pin_memory=true 该为 false。

总结

以上是ag凯发k8国际为你收集整理的warning: leaking caffe2 thread-pool after fork的全部内容,希望文章能够帮你解决所遇到的问题。

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

  • 上一篇:
  • 下一篇:
网站地图