Как запретить TPUEstimator использовать GPU или TPU - PullRequest
Новая
       16

Как запретить TPUEstimator использовать GPU или TPU

0 голосов
/ 06 апреля 2019

Мне нужно заставить TPUEstimator использовать процессор. У меня есть арендованный гугл-компьютер, и на GPU уже идет тренировка. Поскольку процессоры простаивают, я хочу начать второй сеанс Tensorflow для оценки, но я хочу заставить цикл оценки использовать только процессоры, чтобы он не крал время GPU.

Я предполагаю, что в run_config или аналогичном для этого есть флаг, но я не могу найти его в документации TF. 

run_config = tf.contrib.tpu.RunConfig(
    cluster=tpu_cluster_resolver,
    master=FLAGS.master,
    model_dir=FLAGS.output_dir,
    save_checkpoints_steps=FLAGS.save_checkpoints_steps,
    tpu_config=tf.contrib.tpu.TPUConfig(
        iterations_per_loop=FLAGS.iterations_per_loop,
        num_shards=FLAGS.num_tpu_cores,
        per_host_input_for_training=is_per_host))

1 Ответ

0 голосов
/ 26 мая 2019

Вы можете запустить TPUEstimator локально, включив два аргумента: (1) use_tpu должен быть установлен в False и (2) tf.contrib.tpu.RunConfig должен быть передан в качестве аргумента config. 

my_tpu_estimator = tf.contrib.tpu.TPUEstimator(
    model_fn=my_model_fn,
    config=tf.contrib.tpu.RunConfig()
    use_tpu=False)

Большинство примеров моделей TPU можно запустить в локальном режиме, установив флаги командной строки: 

$> python mnist_tpu.py --use_tpu=false --master=''

Дополнительную документацию можно найти здесь .

...