前几天测试官方的stargan-V2,用自己的RTX5000,16G显存,感觉很足够,但是报错误,Out of memory,然后说是需要8G内存,你的显卡12G free,已经使用600M。显然是空间足够使用。网上查找各种解决办法,Batch_size=1,还有 with torch.nn Grad_zero ().等等,试了很多办法,无法解决。
今天突然网上租用服务器,2080ti,然后一次跑通了,至今也不知什么原因,代码没改动,但是两种GPU却有不同结果,哪位大神若是看到了,知道原因可以说一声。感激不尽。我若是知道了也会更新这篇博客。
time:2020-12-10 14:52
使用RTX5000和2080ti的区别测试代码的疑惑?
最新推荐文章于 2024-10-15 11:47:59 发布