GPU内存溢出,以及forward() missing 1 required positional argument: ‘x‘;设置GPU
最新推荐文章于 2024-01-03 18:58:40 发布
在进行深度学习实验时遇到了GPU内存溢出问题,尝试调整batchsize无效。问题根源在于GPU设置代码有误,由于未指定GPU使用,导致多张GPU被启用。解决方案是通过设置环境变量`CUDA_VISIBLE_DEVICES`为'0',只使用一张GPU。此外,还遇到了`forward()`函数缺少参数的问题,可能与批次大小不匹配有关。
摘要由CSDN通过智能技术生成