Function isGPU

tests/common.py:14–15

()

        return("nvidia-smi not found.")

def isGPU():
    return os.path.isfile('/proc/driver/nvidia/version')

gpu_test = unittest.skipIf(not isGPU(), 'Not running GPU tests')
# b/342143152 P100s are slowly being unsupported in new release of popular ml tools such as RAPIDS.