Warning: Leaking Caffe2 thread-pool after fork

詳細(xì)可以參考這里：https://discuss.pytorch.org/t/dataloader-when-num-worker-0-there-is-bug/25643/37

我的情況是在PyTorch 1.9.1，CUDA 11.1， RTX 3090 顯卡上出現(xiàn)這個(gè)問(wèn)題。
解決方法: 將 pin_memory=True 該為 False。

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。