mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-08 18:44:06 +00:00
Install latest tensorflow using pip (#2263)
* Install latest tensorflow using pip * Allow tensorflow test to warn * Fix another way
This commit is contained in:
@@ -1,5 +1,12 @@
|
||||
# Copyright (c) Jupyter Development Team.
|
||||
# Distributed under the terms of the Modified BSD License.
|
||||
import os
|
||||
import sys
|
||||
|
||||
if "NVIDIA_VISIBLE_DEVICES" in os.environ:
|
||||
print("Not running this test in GPU mode")
|
||||
sys.exit(0)
|
||||
|
||||
import tensorflow as tf
|
||||
|
||||
print(tf.constant("Hello, TensorFlow"))
|
||||
|
Reference in New Issue
Block a user