Skip to content
Related Articles

Related Articles

Improve Article
Save Article
Like Article

Python – tensorflow.DeviceSpec.device_type Attribute

  • Last Updated : 28 Jul, 2020

TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning  neural networks. 

device_type is used to get the device type of DeviceSpec.

 Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.  

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning - Basic Level Course

Syntax: tensorflow.DeviceSpec.device_type



Returns: It returns the device type.

Example 1:

Python3




# Importing the library
import tensorflow as tf
  
# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg", device_type ="CPU", device_index = 1)
  
# Printing the DeviceSpec object
print('Device Spec: ', device_spec)
  
# Getting device type
device_type = device_spec.device_type
  
# Printing the result
print('Device Type: ', device_type)

Output:


Device Spec:  <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5ba96af48>
Device Type:  CPU

Example 2:

Python3




# Importing the library
import tensorflow as tf
  
# Initializing Device Specification
device_spec = tf.DeviceSpec(job ="gfg", device_type ="GPU", device_index = 1)
  
# Printing the DeviceSpec object
print('Device Spec: ', device_spec)
  
# Getting device type
device_type = device_spec.device_type
  
# Printing the result
print('Device Type: ', device_type)

Output:


Device Spec:  <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5ba96aee8>
Device Type:  GPU




My Personal Notes arrow_drop_up
Recommended Articles
Page :