HeinrichAD/ProcessorUnit.py

## colab_pu_detector.ipynb

      
        
            
              
              
            

            Raw
  

        
        
          
            
          
          
            
              colab_pu_detector.ipynb
            
          
        
      
    
    

          
    
      
      Sorry, something went wrong. Reload?
      Sorry, we cannot display this file.
      Sorry, this file is invalid so it cannot be displayed.
      
          Viewer requires iframe.
      
    
  
 
    

  

## ProcessorUnit.py
from enum import Enum
from os import environ

class PU(Enum):
  """
  Google Colab Processor Unit / Runtime Type Detector.
  """
  CPU = 1
  GPU = 2
  TPU = 3

  def isCPU():
    return not (PU.isGPU() or PU.isTPU())
  def isGPU():
    return environ["COLAB_GPU"] == '1'
  def isTPU():
    return "TPU_NAME" in environ

  def getPU():
    return PU.GPU if PU.isGPU() else PU.TPU if PU.isTPU() else PU.CPU
	from enum import Enum
	from os import environ

	class PU(Enum):
	"""
	Google Colab Processor Unit / Runtime Type Detector.
	"""
	CPU = 1
	GPU = 2
	TPU = 3

	def isCPU():
	return not (PU.isGPU() or PU.isTPU())
	def isGPU():
	return environ["COLAB_GPU"] == '1'
	def isTPU():
	return "TPU_NAME" in environ

	def getPU():
	return PU.GPU if PU.isGPU() else PU.TPU if PU.isTPU() else PU.CPU