Отдельные скрипты тензорных потоков графических процессоров, работающие на отдельных графических процессорах - PullRequest
1 голос
/ 10 июля 2020

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

I успешно запустил скрипт тензорного потока только на графическом процессоре, но когда я запускаю другой скрипт тензорного потока на другом графическом процессоре, я получаю следующее: 1007 *

Эта ошибка Что происходит? Является ли моя фантазия чисто выдуманной?

РЕДАКТИРОВАТЬ: Кажется, что когда в данный момент запущена программа с интенсивным использованием графических процессоров (например, другой сценарий тензорного потока, видеоигра или что-то в этом роде), возникает первая ошибка. Однако, если скрипт gpu запущен и запущен, а затем запускается видеоигра или другая программа с интенсивным использованием графических процессоров, скрипт замедляется и даже выдает вторую ошибку.

1 Ответ

1 голос
/ 10 июля 2020
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0" #select 0 for first GPU or 1 for second

Поместите этот сценарий в самый верх любого файла python, чтобы вручную назначить графический процессор.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...