search

ardianumam / Tensorflow-TensorRT

This repository is for my YT video series about optimizing a Tensorflow deep learning model using TensorRT. We demonstrate optimizing LeNet-like model and YOLOv3 model, and get 3.7x and 1.5x faster for the former and the latter, respectively, compared to the original models.
303 stars 110 forks source link

Baseline configuration #4

Closed filipski closed 5 years ago

filipski commented 5 years ago

Hi,

Please update the README.md with the description of your environment. I know from your last video that you're running probably on desktop with GTX 1060, but which versions of NVIDIA drivers, CUDA , cuDNN and TensorRT? Similarly, provide you setup for Jetson, please.

ardianumam commented 5 years ago

Hi,

Thanks for pointing it out. Now, it's added.

filipski commented 5 years ago

Thanks! Closing this one then.