tf.test.is_gpu_available
| Просмотреть исходный код на GitHub |
Возвращает, имеет ли TensorFlow доступ к GPU. (устарело)
tf.test.is_gpu_available(
cuda_only=False, min_cuda_compute_capability=None
)
| Аргументы | |
|---|---|
cuda_only | ограничить поиск графическими процессорами CUDA. |
min_cuda_compute_capability | пара (major, minor), указывающая минимальную требуемую вычислительную способность CUDA, или None, если требований нет. |
Обратите внимание, что имя ключевого аргумента "cuda_only" вводящее в заблуждение (так как функция вернёт true, когда доступен графический процессор независимо от того, был ли TF скомпилирован с поддержкой CUDA или ROCm. Однако изменений здесь нет, потому что
++ Изменение имени "cuda_only" на что-то более общее нарушит обратную совместимость
++ Добавление эквивалента "rocm_only" потребует проверки типа сборки в реализации. Это, в свою очередь, потребовало бы того же для CUDA и, следовательно, потенциально нарушило бы обратную совместимость
++ Добавление нового "cuda_or_rocm_only" не нарушит обратную совместимость, но потребует, чтобы большинство (если не все) вызывающие функции обновили вызов, чтобы использовать "cuda_or_rocm_only" вместо "cuda_only"
| Возвращаемое значение | |
|---|---|
| True, если доступен графический процессор требуемого типа. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/test/is_gpu_available
