Sign inSign up

jgato/tensorflow-jetson

By jgato

Updated about 6 years ago

Docker image for NVIDIA Jetson Nano and Tensorflow. It provides TF 2.2

Image
0

172

jgato/tensorflow-jetson repository overview

Overview:

  • The host should a Jetson Nano with Jetpack 43 and L4T Kernel drivers 32.4.3
  • It uses NVIDIA L4T base image 32.4.3
  • The image provides TF2.2, which is compatible with the host providing CUDA 10.2

Usage

$ docker run -it jgato/tensorflow-jetson:R32.4.3-TF-2.2 /bin/bash
root@e139ae535f9b:/# python3
Python 3.6.9 (default, Jul 17 2020, 12:50:27) 
[GCC 8.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow.python.client.device_lib as tf
>>> tf.list_local_devices()
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 7685036805777320728
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 14014663960560489866
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 11504966275169859351
physical_device_desc: "device: XLA_GPU device"
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 170045440
locality {
  bus_id: 1
  links {
  }
}
incarnation: 9320175265347385001
physical_device_desc: "device: 0, name: NVIDIA Tegra X1, pci bus id: 0000:00:00.0, compute capability: 5.3"
]

Tag summary

Content type

Image

Digest

Size

1.2 GB

Last updated

about 6 years ago

docker pull jgato/tensorflow-jetson:R32.4.3-TF-2.2