close
網站架設
近重新調試一段pytorch 程式碼,之前的伺服器上完全沒問題,但換了一台機器,從頭安裝了新版本的cuda, anaconda ,pytorch 等,之前的程式碼呈現各類版本不合適的問題。
問題:
目前說說這個問題。運行pytorch 時出現的情況以下:
緣由:
在網上查找效果,緣由是多進程的原因。具體可參考博文:Python 中的if __name__ == '__main__' 該若何理解。
解決方式:
既然是多履行緒的緣由,那麼可以從兩個角度解決問題:
1. 程式碼在運行 epoch 之前,加上 if __name__=='__main__'
試過,有用,一次經由過程。
2. 不利用多線程,也就是去掉num_workers 參數,或設定 num_workers=0。
(目前還沒試過,因為我照舊想用多履行緒。)
文章出處: NetYea 網頁設計
參考文章
https://blog.csdn.net/u014546828/article/details/109235539
文章標籤
全站熱搜