tf.test.is_gpu_available

Возвращает, может ли TensorFlow получить доступ к GPU. (устарела)

Посмотреть псевдонимы

Псевдонимы для совместимости при миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.test.is_gpu_available

tf.test.is_gpu_available(
    cuda_only: bool = False,
    min_cuda_compute_capability: Optional[tuple[int, int]] = None
) -> bool

Например,

>>> gpu_available = tf.test.is_gpu_available()
>>> is_cuda_gpu_available = tf.test.is_gpu_available(cuda_only=True)
>>> is_cuda_gpu_min_3 = tf.test.is_gpu_available(True, (3,0))

Аргументы
`cuda_only`	ограничивает поиск GPU с поддержкой CUDA.
`min_cuda_compute_capability`	пара (major,minor), указывающая минимальную требуемую вычислительную способность CUDA, или None, если ограничений нет.

Обратите внимание, что имя аргумента-ключевого слова "cuda_only" вводит в заблуждение (поскольку процедура вернёт true, если доступно устройство GPU, независимо от того, была ли TF собрана с поддержкой CUDA или ROCm). Однако здесь ничего не меняем, потому что

++ Изменение имени "cuda_only" на более универсальное нарушило бы обратную совместимость

++ Добавление эквивалентного аргумента "rocm_only" потребовало бы от реализации проверки типа сборки. В свою очередь, это потребовало бы сделать то же самое для CUDA и могло бы нарушить обратную совместимость

++ Добавление нового аргумента "cuda_or_rocm_only" не нарушило бы обратную совместимость, но потребовало бы от большинства (если не от всех) вызывающих функций изменить вызов и использовать "cuda_or_rocm_only" вместо "cuda_only"

Возвращает
True, если доступно устройство GPU запрошенного типа.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/test/is_gpu_available