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