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

最低0.47元/天 解锁文章
1万+

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



