Tensorflow GPU

Hello, i’m trying to run tensorflow on Jetson TX2 and when i try the test this is the result:

nvidia@tegra-ubuntu:~$ python3
Python 3.5.2 (default, Nov 12 2018, 13:43:14)
[GCC 5.4.0 20160609] on linux
Type “help”, “copyright”, “credits” or “license” for more information.

import tensorflow
hello = tensorflow.constant(‘Hello, Tensorflow!’)
sess = tensorflow.Session()
2019-02-27 11:22:33.356274: I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:864] ARM64 does not support NUMA - returning NUMA node zero
2019-02-27 11:22:33.356409: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1392] Found device 0 with properties:
name: NVIDIA Tegra X2 major: 6 minor: 2 memoryClockRate(GHz): 1.3005
pciBusID: 0000:00:00.0
totalMemory: 7.66GiB freeMemory: 2.77GiB
2019-02-27 11:22:33.356460: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1471] Adding visible gpu devices: 0
2019-02-27 11:22:34.324949: I tensorflow/core/common_runtime/gpu/gpu_device.cc:952] Device interconnect StreamExecutor with strength 1 edge matrix:
2019-02-27 11:22:34.325027: I tensorflow/core/common_runtime/gpu/gpu_device.cc:958] 0
2019-02-27 11:22:34.325055: I tensorflow/core/common_runtime/gpu/gpu_device.cc:971] 0: N
2019-02-27 11:22:34.325193: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1084] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 2243 MB memory) -> physical GPU (device: 0, name: NVIDIA Tegra X2, pci bus id: 0000:00:00.0, compute capability: 6.2)

print(sess.run(hello))
b’Hello, Tensorflow!’

So, it works, but my real question is, am i using GPU?

Thanks

Hi,

You can enable the device placement log to check how TensorFlow execute your model:

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

Thanks.