diff options
Diffstat (limited to 'tensorflow/contrib/eager/python/examples/notebooks/1_basics.ipynb')
-rw-r--r-- | tensorflow/contrib/eager/python/examples/notebooks/1_basics.ipynb | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/eager/python/examples/notebooks/1_basics.ipynb b/tensorflow/contrib/eager/python/examples/notebooks/1_basics.ipynb index 01616f2e7d..459f2f4a7d 100644 --- a/tensorflow/contrib/eager/python/examples/notebooks/1_basics.ipynb +++ b/tensorflow/contrib/eager/python/examples/notebooks/1_basics.ipynb @@ -429,7 +429,9 @@ "cpu_tensor = tf.random_normal([SIZE, SIZE])\n", "\n", "if is_gpu_available:\n", - " gpu_tensor = cpu_tensor.gpu()" + " gpu_tensor = cpu_tensor.gpu()\n", + "else:\n", + " print(\"GPU not available.\")" ] }, { |