本文分享了一次解决CUDA运行时内存不足的问题经验。作者最初怀疑是CUDA配置问题，尝试了网上推荐的清除缓存方法，但未奏效。最终发现是代码中设置的GPU内存分配比例过高导致，将‘per_process_gpu_memory_fraction’参数从0.9调整至0.8后，问题得以解决。
之前还可以运行之后不行了还甚至怀疑自己cuda有问题，用了网上的方法sudo rm -rf ~/.nv/还是不行，其实就是内存不够，找到自己代码
‘per_process_gpu_memory_fraction’: 0.9,
改0.9为0.8就好了
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
