CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 6.00 GiB total capacity; 4.54 GiB already allocated; 14.94 MiB free; 4.64 GiB reserved in total by PyTorch)
分析问题 CUDA内存超载
解决尝试一:GPU未被调用,

实际上有那么一瞬间调用了。
按照上面的方法调用,还是这样。
我感觉主要问题是很多内存没有被调用,
上面的报错翻译是CUDA内存不足。尝试分配16.00 MiB(GPU 0;总容量6.00 GiB;4.54已分配的GiB;14.94无MiB;PyTorch共预留4.64个GiB)
我想到很多内存没有调用
于是查询了

博客内容涉及CUDA内存不足的问题,用户尝试解决时发现GPU未被充分利用,且CUDA只识别到一个GPU,尽管系统有多个。通过检查和设置`CUDA_VISIBLE_DEVICES`环境变量,确认只有一个GPU可供使用。解决方案可能包括优化内存分配和使用多GPU策略。
&spm=1001.2101.3001.5002&articleId=120060303&d=1&t=3&u=04426604a71f4a65b51acb03f731364b)
6657

被折叠的 条评论
为什么被折叠?



