windows下用anaconda安裝的pytorch only cpu版本訓練時只用了一個cup核

torch.set_num_threads(35)

用這句話可以設置線程數量,用設備管理器看自己電腦有多少個線程然後設置比總數稍微少一點就可以了。

訓練速度比之前快很多,沒有gpu就只能這樣了。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章