pytorch中出現RuntimeError: CUDA out of memory.

1、原因時多線程配置的問題,你將num_workers設置爲0即可解決.

2、模型太大。

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