- 博客(3)
- 收藏
- 关注
RSS订阅原创 paddlepaddle-gpu安装报错：（The GPU architecture in your current machine is Pascal, which ......）
后来进行问题查找，cuda 120，paddle版本 2.6.0.post120对显卡算力的架构有一定要求。我的显卡（算力6.1）暂时不满足。（paddle版本2.5.1需要7.0以上算力的架构）我刚开始以为这只是一个普通警告信息，所以忽略了。但是后面进行模型训练时，发现 loss 一直是 nan，模型无法进行正常训练。刚开始，我以为是学习率、batch_size参数设置有问题，经过不同参数的调整，问题还是无法解决。
2024-02-23 14:37:45
853
3
原创 关于使用paddlepaddle-gpu训练模型时，发生错误：(Out of memory error on GPU 0. Cannot allocate 49.000000MB ......）
显存不足报错，尝试减少下batch_size、或者裁剪图片等降低下显存占用。或者你不要在本地跑，用高配的服务器去训练后导出模型参数再考虑本地做预测及部署。我尝试减少batch_size=8（原：64）后，成功运行！
2024-02-22 23:03:59
879
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人