search

Technica-Corporation / Tegra-Docker

Instructions and key files to enable Docker support on NVIDIA Tegra devices, specifically the TX-2.
Apache License 2.0
234 stars 61 forks source link

TensorRT in Docker using Xavier #14

Closed statikkkkk closed 5 years ago

statikkkkk commented 5 years ago

Hi,

I am trying to get TensorRT working with Xavier. The provided tx2-docker wrapper works perfectly with Tensorflow until I tried to get TensorRT working w/ Tensorflow.

System version:

The error I got is:

2019-06-20 22:49:25.752103: F tensorflow/contrib/tensorrt/log/trt_logger.cc:42] DefaultLogger Assertion failed: eglCreateStreamKHR != nullptr
dla/eglUtils.cpp:61
Aborting...

I am able to run the script perfectly fine outside Docker, but have this issue if running with Docker.

statikkkkk commented 5 years ago

Closed due to official support for GPU Docker in JetPack is supported.

https://developer.nvidia.com/embedded/jetpack

From Nvidia's website:

JetPack 4.2.1 also introduced two beta features: NVIDIA Container Runtime with Docker integration and TensorRT support for INT-8 DLA operations.