Installation¶
Docker¶
Latest TensorFlow 2.x docker image from NGC is recommended.
Clone the tensorflow-quantization
repository, pull the docker image, and launch the container.
$ cd ~/ $ git clone https://github.com/NVIDIA/TensorRT.git $ docker pull nvcr.io/nvidia/tensorflow:22.03-tf2-py3 $ docker run -it --runtime=nvidia --gpus all --net host -v ~/TensorRT/tools/tensorflow-quantization:/home/tensorflow-quantization nvcr.io/nvidia/tensorflow:22.03-tf2-py3 /bin/bash
After the last command, you will be placed in the /workspace
directory inside the running docker container, whereas the tensorflow-quantization
repository is mounted in the /home
directory.
$ cd /home/tensorflow-quantization $ ./install.sh $ cd tests $ python3 -m pytest quantize_test.py -rP
If all tests pass, installation is successful.
Local¶
$ cd ~/
$ git clone https://github.com/NVIDIA/TensorRT.git
$ cd TensorRT/tools/tensorflow-quantization
$ ./install.sh
$ cd tests
$ python3 -m pytest quantize_test.py -rP
If all tests pass, installation is successful.