330
edits
Changes
Jump to navigation
Jump to search
→Using Tensorflow
A quick test of whether tensorflow-gpu is working for ''python3.6'', type the following into a terminal:
python3.6 -c "import tensorflow as tf; sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))"
This will report back which CPU and GPU devices the tensorflow is using. If there is no information for the GPU device, there is something wrong.
