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

猜你喜欢

转载自blog.csdn.net/u011622208/article/details/106004299