Unable to register cuBLAS factory

[srikanth-ml-dev]

Please can anyone help me out with the issue. Thanks in Advance

[quic-hitameht]

Hi @srikanth-ml-dev

Could you please provide more details on what steps you have performed so far. Thank you.

[srikanth-ml-dev]

Hi @quic-hitameht
I just followed all the steps that were in the official documentation of aimet tensorflow gpu. I tried with cuda 11.4 and 11.2.
I could solve the issue just by just upgrading from tensorflow 2.10 to tensorflow 2.11. cuBLAS error somehow has been fixed in tensorflow 2.11. But aimet version 1.30.0 demands tensorflow-gpu 2.10. As of now I have managed to get rid of the issue

[srikanth-ml-dev]

Hi @quic-hitameht

Can you please help me solve the error in the image above