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, когда устройство GPU доступно независимо от того, был ли TF скомпилирован с поддержкой CUDA или ROCm. Однако никаких изменений здесь не делается, потому что
++ Изменение имени «cuda_only» на что-то более общее нарушит обратную совместимость
++ Добавление эквивалентного «rocm_only» потребует проверки типа сборки в реализации. Это, в свою очередь, потребовало бы того же для CUDA и, следовательно, потенциально могло бы нарушить обратную совместимость
++ Добавление нового «cuda_or_rocm_only» не нарушит обратную совместимость, но потребует, чтобы большинство (если не все) вызывающие стороны обновили вызов, чтобы использовать «cuda_or_rocm_only» вместо «cuda_only»
| Возвращаемое значение | |
|---|---|
| True, если устройство GPU требуемого типа доступно. |
© 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/r1.15/api_docs/python/tf/test/is_gpu_available
