Python test_util.IsGoogleCudaEnabled方法代碼示例

本文整理匯總了Python中tensorflow.python.framework.test_util.IsGoogleCudaEnabled方法的典型用法代碼示例。如果您正苦於以下問題：Python test_util.IsGoogleCudaEnabled方法的具體用法？Python test_util.IsGoogleCudaEnabled怎麽用？Python test_util.IsGoogleCudaEnabled使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類tensorflow.python.framework.test_util的用法示例。

在下文中一共展示了test_util.IsGoogleCudaEnabled方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: is_built_with_cuda

# 需要導入模塊: from tensorflow.python.framework import test_util [as 別名]
# 或者: from tensorflow.python.framework.test_util import IsGoogleCudaEnabled [as 別名]
def is_built_with_cuda():
  """Returns whether TensorFlow was built with CUDA (GPU) support."""
  return _test_util.IsGoogleCudaEnabled()

開發者ID:ryfeus，項目名稱:lambda-packs，代碼行數:5，代碼來源:test.py

示例2: testIsGoogleCudaEnabled

# 需要導入模塊: from tensorflow.python.framework import test_util [as 別名]
# 或者: from tensorflow.python.framework.test_util import IsGoogleCudaEnabled [as 別名]
def testIsGoogleCudaEnabled(self):
    # The test doesn't assert anything. It ensures the py wrapper
    # function is generated correctly.
    if test_util.IsGoogleCudaEnabled():
      print("GoogleCuda is enabled")
    else:
      print("GoogleCuda is disabled")

開發者ID:tobegit3hub，項目名稱:deep_image_model，代碼行數:9，代碼來源:test_util_test.py

注：本文中的tensorflow.python.framework.test_util.IsGoogleCudaEnabled方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台，相關代碼片段篩選自各路編程大神貢獻的開源項目，源碼版權歸原作者所有，傳播和使用請參考對應項目的License；未經允許，請勿轉載。