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。

總結

以上是生活随笔為你收集整理的Warning: Leaking Caffe2 thread-pool after fork的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。