GPU很空闲总是提示GPU out of memeory，可能是因为Docker的原因

最新推荐文章于 2024-08-09 00:38:37 发布

努力搞学术的小刘

最新推荐文章于 2024-08-09 00:38:37 发布

阅读量165

点赞数 3

文章标签： YOLO docker 深度学习

本文链接：https://blog.csdn.net/weixin_46647483/article/details/140069570

版权

我遇到的具体问题是，当我完全重新开机后，运行模型后不会出现GPU out of memeory，但是当我终止后再次运行却会如此报错。
然后我就进行百度发现网络上大部分的解决办法是：

调小参数的个数
或者更改更小的模型
压缩输入图像的大小
这三个办法是比较主流的原因，但是我在进行尝试后都没有解决。反而是，在不断试错过程中发现，只要我打开Docker后Pycharm就不能正常运行，如果关闭Docker一切都可以恢复正常。
目前，还没有完全解决这个问题。如果你是运用本地Python运行的话，可以尝试关闭Docker后再运行，从而检查是否还会报错。