warning: leaking caffe2 thread-凯发ag旗舰厅登录网址下载

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旗舰厅登录网址下载为你收集整理的warning: leaking caffe2 thread-pool after fork的全部内容，希望文章能够帮你解决所遇到的问题。

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