1、刚开始我以为是batch_size设置问题导致内存溢出,但是仔细观察发现空闲内存大于需要使用的内存,
解决办法
卸载torch 当前torch 版本为1.6.0 torchvision版本为0.7.0
回退到 torch 1.2.0 torchvision 0.4.0
问题解决
显卡内存足够但是torch报错RuntimeError: CUDA out of memory
最新推荐文章于 2024-06-13 23:10:33 发布
1、刚开始我以为是batch_size设置问题导致内存溢出,但是仔细观察发现空闲内存大于需要使用的内存,
解决办法
卸载torch 当前torch 版本为1.6.0 torchvision版本为0.7.0
回退到 torch 1.2.0 torchvision 0.4.0
问题解决