Merge pull request #3141 from yaroslavvb/patch-1

assert fails on requesting gpus when not available

Merge pull request #3141 from yaroslavvb/patch-1
assert fails on requesting gpus when not available
1f34fcaf · Toby Boyd · GitHub · 5fa2a4e6 · 6b9d5fba · 1f34fcaf
Unverified Commit 1f34fcaf authored Jan 24, 2018 by Toby Boyd Committed by GitHub Jan 24, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tutorials/image/cifar10_estimator/cifar10_main.py tutorials/image/cifar10_estimator/cifar10_main.py +2 -0

No files found.
--- a/tutorials/image/cifar10_estimator/cifar10_main.py
+++ b/tutorials/image/cifar10_estimator/cifar10_main.py
@@ -503,6 +503,8 @@ if __name__ == '__main__':
      help='Epsilon for batch norm.')
  args = parser.parse_args()

+  if args.num_gpus > 0:
+    assert tf.test.is_gpu_available(), "Requested GPUs but none found."
  if args.num_gpus < 0:
    raise ValueError(
        'Invalid GPU count: \"--num-gpus\" must be 0 or a positive integer.')