我用train_tripletloss.py去训练模型，默认是用CPU训练，我怎么设置成用GPU训练呢

来源：9-11 facenet模型训练

易水风行0

2019-07-17

我用train_tripletloss.py去训练模型，默认是用CPU训练，我怎么设置成用GPU训练呢
图片描述

写回答

2回答

快乐冰冰bing

2019-12-17

我之前也是遇到这样的问题，我是有一步没有按照github教程上做，就是没有在research/下用pip3 install tensorflow-gpu==1.12.0 安装tensorfow ，所以我也是刚开始用的是cpu 。如果对你有帮助，采纳下回答，谢谢，祝你学习愉快～～

会写代码的好厨师

2019-07-25

确认一下，你的tensorflow是cpu版本的还是gpu. 版本的。

Python3+TensorFlow打造人脸识别智能小程序

理论与实战项目双管齐下，让AI技术真正落地应用，适合毕设展示。

1086 学习 · 538 问题

相似问题

回答 1

回答 3

回答 1

回答 1

回答 1