【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没有编译成功,所以没有测试。

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