2401_82920456 2024-12-04 11:04 采纳率: 0%
浏览 28

请问在低算力显卡中如何关闭amp自动混淆精度模式?

初次接触飞桨,部署手写识别demo时,

model.fit(train_dataset, epochs=5, batch_size=64, verbose=1)行出现UserWarning: For float16, amp only support NVIDIA GPU with Compute Capability 7.0 or higher, current GPU is: NVIDIA GeForce GTX 960, with Compute Capability: 5.2.错误。提示我amp模式下需要7以上的算力,我的gpu算力只有5.2  。

请问amp可以关闭吗?能否在地算力情况下使用gpu进行测试?希望懂的能帮忙解惑。在不升级gpu的情况下,是否还能使用飞桨进行模型训练呢?

  • 写回答

1条回答 默认 最新

  • 微雨盈萍cbb 2024-12-11 14:17
    关注

    use_amp=False就可以了

    评论

报告相同问题?

问题事件

  • 创建了问题 12月4日