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/r2.3/api_docs/python/tf/test/is_gpu_available
