No module named quantization

NVIDIA-AI-IOT / cuDLA-samples

YOLOv5 on Orin DLA

Other

167 stars 17 forks source link

No module named quantization #30

Open zyitom opened 4 months ago

zyitom commented 4 months ago

2024-03-13_11-11 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 commented 3 months ago

@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