Tensorflow(TF) GPU installation

[saiksaketh]

Hi,

https://quic.github.io/aimet-pages/releases/1.35.0/install/index.html#release-packages

In the above link, in TF section, it says

Tensorflow 2.10 GPU with CUDA 11.x

python3 -m pip install https://github.com/quic/aimet/releases/download/1.35.0/aimet_tensorflow-1.35.0.cu118-cp310-cp310-manylinux_2_34_x86_64.whl

Does it mean my environment should have Tensorflow 2.10 GPU with CUDA 11.x non-negotiable ?, as I am working on TF 2.15 and CUDA 12.x

Installation is not straight forward either, any help ?

[saiksaketh]

I get a consistent aimet_xx.whl is not a supported wheel on this platform.

[quic-klhsieh]

@saiksaketh , using the above pip install command I'm able to install aimet_tensorflow for CUDA 12.x at least. Would you be able to post any debug log or verbose output you get when pip install fails?