BrokenPipeError: [Errno 32] Broken pipe


運行Pytorch tutorial代碼報錯:BrokenPipeError: [Errno 32] Broken pipe

源代碼地址: Training a classifier (CIFAR10)

該問題的產生是由於windows下多線程的問題,和DataLoader類有關,具體細節點這里Fix memory leak when using multiple workers on Windows

解決方案:

    修改調用torch.utils.data.DataLoader()函數時的 num_workers 參數。該參數官方API解釋如下: 

  • num_workers (int, optional) – how many subprocesses to use for data loading. 0 
    means that the data will be loaded in the main process. (default: 0)

    該參數是指在進行數據集加載時,啟用的線程數目。截止當前2018年5月9日11:15:52,如官方未解決該BUG,則可以通過修改num_works參數為 ,只啟用一個主進程加載數據集,避免在windows使用多線程即可。

轉自:https://blog.csdn.net/u014380165/article/details/79058479


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM