Tensorflow: Posso exportar um modelo treinado pela GPU para uma máquina somente CPU para exibição / avaliação?

Criado em 16 jul. 2016  ·  3Comentários  ·  Fonte: tensorflow/tensorflow

Treinei um modelo em uma máquina GPU.
Quando carrego esse modelo em uma máquina somente cpu, obtenho o erro abaixo.

Obrigado por qualquer sugestão!

awaiting tensorflower

Comentários muito úteis

Além disso, no futuro, StackOverflow é o lugar certo para fazer essas perguntas.

Todos 3 comentários

@vrv Você poderia dar uma olhada nisso, por favor? Temos esse apoio agora? Obrigado.

Sim, você pode passar tf.ConfigProto (allow_soft_placement = True) para tf.Session se realmente quiser ignorar as diretivas de posicionamento de dispositivo no gráfico. Não sou um grande fã de usar essa opção (prefiro que o gráfico seja reescrito para remover os campos do dispositivo explicitamente), mas isso deve funcionar.

Além disso, no futuro, StackOverflow é o lugar certo para fazer essas perguntas.

Esta página foi útil?
0 / 5 - 0 avaliações