TensorFlow 1 version | View source on GitHub |
Returns whether TensorFlow was built with GPU (i.e. CUDA or ROCm) support.
tf.test.is_built_with_gpu_support()
TensorFlow 1 version | View source on GitHub |
Returns whether TensorFlow was built with GPU (i.e. CUDA or ROCm) support.
Compat aliases for migration
See Migration guide for more details.
tf.test.is_built_with_gpu_support()
此页内容是否对您有帮助
感谢反馈!