Тензор потока 2.0 обнаружит только один графический процессор из 2 графических процессоров - PullRequest
0 голосов
/ 30 апреля 2020

у меня 2 гпус cuda versionL 10.0

gpus = tf.config.experimental.list_physical_devices('GPU')
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

но это Когда я набираю этот тензор потока, обнаруживается только одно устройство

os.environ["CUDA_VISIBLE_DEVICES"]="1"
tf.test.gpu_device_name()

на самом деле это не работает

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