моя настройка номер устройства GPU не работает? - PullRequest
0 голосов
/ 01 марта 2020
import keras
import tensorflow as tf
print(keras.__version__)
print(tf.__version__)

2.3.1 1.15.0

Я пытаюсь настроить GPU на номер 1

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" 
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
os.environ["CUDA_VISIBLE_DEVICES"]

но оно не получает 1-е устройство, оно всегда получает 0-е устройство это правильный способ настроить устройство GPU? Вам нужна другая информация?

...