Как разместить узлы на тензорном виртуальном GPU? - PullRequest
Новая
       25

Как разместить узлы на тензорном виртуальном GPU?

0 голосов
/ 01 мая 2020

У меня есть один графический процессор, я установил конфигурацию сеанса как: 

gpu_options = tf.GPUOptions(
    visible_device_list='0',
    experimental=tf.GPUOptions.Experimental(virtual_devices=[
        tf.GPUOptions.Experimental.VirtualDevices(
            memory_limit_mb=[5172, 5172])]))
tf_config = tf.ConfigProto(gpu_options = gpu_options, device_count = {'GPU':1})
tf_config.log_device_placement=True

Чтобы я мог создать два виртуальных графических процессора, которые может использовать сессия: 

Device mapping:
/job:localhost/replica:0/task:0/device:XLA_CPU:0 -> device: XLA_CPU device
/job:localhost/replica:0/task:0/device:XLA_GPU:0 -> device: XLA_GPU device
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: TITAN Xp, pci bus id: 0000:08:00.0, compute capability: 6.1
/job:localhost/replica:0/task:0/device:GPU:1 -> device: 0, name: TITAN Xp, pci bus id: 0000:08:00.0, compute capability: 6.1

Однако все узлы в график сеанса помещается в / job: localhost / реплика: 0 / task: 0 / устройство: CPU: 0 и / job: localhost / replica: 0 / task: 0 / device: GPU: 0, с / job: localhost / replica : 0 / задача: 0 / устройство: GPU: 1 бесполезно. Как разрешить сеансу использовать оба виртуальных графических процессора?

...