No module named quantization

[zyitom]

I've following the instructions and download the pytorch-quantization via '''pip install pytorch-quantization --extra-index-url https://pypi.ngc.nvidia.com''' this link but when i run the qat.py it met the error and said no modules found ,what shoule i do

[liuanqi-libra7]

@zyitom It seems that the library installation was not successful. You can try using the following command to install it.

$ git clone -b v8.6.1 https://github.com/NVIDIA/TensorRT.git
$ cd TensorRT/tools/pytorch-quantization
$ python setup.py install

[lynettez]

closing since no activity for several months, thanks all!