quic / aimet

AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.
https://quic.github.io/aimet-pages/index.html
Other
2.09k stars 374 forks source link

Installation problem #3210

Open kokhazade opened 1 month ago

kokhazade commented 1 month ago

Hello

I am trying to install AIMET from release package using instructions in this link. I installed prerequisite packages and set the following:

export AIMET_VARIANT=tf_gpu
export release_tag=1.33.0
export download_url="https://github.com/quic/aimet/releases/download/${release_tag}"
export wheel_file_suffix="cp310-cp310-linux_x86_64.whl"

Then after running command python3 -m pip install ${download_url}/aimet_tensorflow-${AIMET_VARIANT}_${release_tag}-${wheel_file_suffix} I get the following error:

ERROR: Invalid requirement: 'aimet-tensorflow==tf-gpu-1.33.0': Expected end or semicolon (after name and no valid version specifier)
    aimet-tensorflow==tf-gpu-1.33.0
                    ^

I tried to change the wheel_file_suffix based on this link filenames but I get the same error. Which part should change to get it working?

ChloeEHKim commented 1 month ago

Downgrade pip version fixed the issue in my case. I initially had pip version of 24.2 and I lowered it down to 23.0.1 python -m pip install pip==23.0.1

songh11 commented 1 month ago

Downgrade pip version fixed the issue in my case. I initially had pip version of 24.2 and I lowered it down to 23.0.1 python -m pip install pip==23.0.1

This also solves the aimet torch installation problem.