March 2019
Intermediate to advanced
576 pages
20h 51m
Portuguese (Portugal, Brazil)
Content preview from Mãos à Obra: Aprendizado de Máquina com Scikit-Learn e TensorFlow
Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
Start your free trial
330 | Capítulo 12: Distribuindo o TensorFlow Por Dispositivos e Servidores
Programa #1 Programa #2
Figura 12-3. Cada programa pega duas GPUs para si
Outra opção seria dizer ao TensorFlow que capture apenas uma fração da memória. Por
exemplo, crie um objeto ConfigProto para fazer o TensorFlow capturar apenas 40% da
memória de cada GPU, ajuste sua opção
gpu_options.per_process_gpu_memory_frac-
tion para 0.4 e crie a sessão utilizando esta configuração:
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4
session = tf.Session(config=config)
Agora dois programas como este podem rodar em paralelo usando as mesmas GPU (mas ...