初次接触飞桨,部署手写识别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的情况下,是否还能使用飞桨进行模型训练呢?