Deep Learning/TensorRT

[TensorRT] TensorFlow Container to accelerate via TF-TRT

jstar0525 2022. 1. 23. 04:02
반응형

If you have Docker 19.03 or later, a typical command to launch the container is:

$ docker run --gpus all -it --rm -v local_dir:container_dir nvcr.io/nvidia/tensorflow:xx.xx-tfx-py3

If you have Docker 19.02 or earlier, a typical command to launch the container is:

$ nvidia-docker run -it --rm -v local_dir:container_dir nvcr.io/nvidia/tensorflow:xx.xx-tfx-py3

 

https://catalog.ngc.nvidia.com/orgs/nvidia/containers/tensorflow

 

TensorFlow | NVIDIA NGC

TensorFlow is an open source platform for machine learning. It provides comprehensive tools and libraries in a flexible architecture allowing easy deployment across a variety of platforms and devices.

catalog.ngc.nvidia.com

 

 

https://docs.nvidia.com/deeplearning/frameworks/support-matrix/index.html

 

Frameworks Support Matrix :: NVIDIA Deep Learning Frameworks Documentation

 

docs.nvidia.com

 

반응형