TensorFlow

API

 tf.test / is_built_with_cuda


Returns whether TensorFlow was built with ROCm (GPU) support.


此页内容是否对您有帮助