【pytorch】 GPU顯存充足out of memory解決辦法

1. 問題

在4塊2080ti的主機上跑centernet,採用的是pytorch0.4.1。報 cuda error,out of memory

2. 解決

在pytorch1.2.0上不會出現以上問題。1.1.0因爲centernet的DCNv2沒有編譯成功,所以沒有測試。

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