如何选择在tensorflow中使用带有tf.device()的GPU

时间:2016-09-28 15:35:47

标签: tensorflow deep-learning

我从models / cifar10 / cifar10_train.py运行tensorflow示例,在示例脚本中,他们说它将使用一个GPU,但我确实发现他们定义使用GPU与tf.device('/ gpu:0')一样): 并且他们用CPU定义存储一些中间变量,所以我想看看tensorflow如何处理GPU和CPU?使用GPU是默认选择??? 以下是该示例的链接: https://www.tensorflow.org/versions/r0.10/tutorials/deep_cnn/index.html 任何建议将被认真考虑, 提前致谢 美好的一天

0 个答案:

没有答案