为了使用contrib.learn.Estimator进行多GPU训练,我试图在我的model_fn中指定GPU分配。 在伪代码: def model_fn(X, y):
with tf.device('/gpu:1'):
... various tensorflow ops for model ...
return predictions, loss, tra
我想学习与的张量流用于学习与TensorFlow高级API和无法获得可重复的结果的问题。 TensorFlow 0.12.0-RC0(CPU只) 蟒蛇3.5 import numpy as np
import tensorflow as tf
import os
import shutil
import random
#print(tf.__version__)
#0.12.0-rc