爲了使用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