【pytorch】由dataloader引發的死鎖

問題闡述: pytorch在訓練過程中由於dataloader 多進程加載數據 導致的死鎖, 使得程序訓練暫停。該問題在pytorch的官方github上的issue上有提到, 這裏不再贅述。 https://github.com/pytorch/pytorch/issues/1355 問題出現的平臺 我這邊出現的平臺,不限於 pytorch 1.1 python 3.6 CUDA 9.0 問題解
相關文章
相關標籤/搜索