TX2 GPU неправильно распределяет память - PullRequest
2 голосов
/ 29 апреля 2019

Я пытаюсь запустить код на TX2, но код тензорного потока, который распределяет использование памяти графическим процессором, кажется, работает странным образом.

Вот код, который я должен выделить для памяти:

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.0
config.gpu_options.visible_device_list = "0"
set_session(tf.Session(config=config))

Странно то, что когда я использую 0.0 вместо 0.5, обработка происходит быстрее. И когда я использую 0.9, я получаю следующую ошибку:

tenorflow.python.framework.errors_impl.InternalError: сбой синхронизации графического процессора

Что здесь происходит?

...