TensorFlow

API

 tf.test / gpu_device_name


Returns whether TensorFlow was built with GPU (i.e. CUDA or ROCm) support.


此页内容是否对您有帮助