win10+mx150深度学习GPU加速
使用keras调用GPU加速十分简便,keras自动调用,不用命令
建议安装conda,比pip好用很多
直接conda install tensorflow-gpu
conda 会自动安装适配的cuda
可以看到加速效果十分明显,原先一个epoch 260s,现在只要17秒,非常棒
直接用tensorflow的调用方法后续研究
使用小米13.3i7独显笔记本
Epoch 1/30
- 18s - loss: 0.4144 - acc: 0.8683 - val_loss: 0.0697 - val_acc: 0.9779
Epoch 2/30
- 17s - loss: 0.1280 - acc: 0.9621 - val_loss: 0.0417 - val_acc: 0.9879
Epoch 3/30
- 17s - loss: 0.0968 - acc: 0.9716 - val_loss: 0.0397 - val_acc: 0.9874
Epoch 4/30
- 17s - loss: 0.0805 - acc: 0.9763 - val_loss: 0.0304 - val_acc: 0.9912