"No module named 'keras'" reported in running tao_byom --help

[xutao3]

I have just followed the steps in BYOM Converter - NVIDIA Docs, however when I run “tao_byom --help” to verify the installation, the error “No module named ‘keras’” was reported, however I can import keras in python console, everything seems fine. What’s the problem?

Reported Error:
(byom) jzyq@jzyq-PCServer:~$ tao_byom --help
Traceback (most recent call last):
File “/home/jzyq/.local/bin/tao_byom”, line 5, in
from tao_byom.entrypoint.entrypoint import main
File “/home/jzyq/.local/lib/python3.8/site-packages/tao_byom/entrypoint/entrypoint.py”, line 8, in
import keras
ModuleNotFoundError: No module named ‘keras’

[Morganh]

Please refer to my comment and log shared in Bug? How to solve errors when I try to install TAO BYOM Converter TF2? - #6 by Morganh.

[xutao3]

Thank you for your information. Downgrading the numpy to 1.21 cannot solve my problem.

My problem is the keras, and it is very strange that the keras has been installed, and I can use it in python console, but when tao_byom calls it, the keras problem happens.

[Morganh]

You can follow my steps and retry. https://forums.developer.nvidia.com/uploads/short-url/cWKBuxsTUEbQgUrKb1Em5fil811.txt.

[xutao3]

I can run tao_byom correctly on my another PC with completely same procedure, don’t know the reason, but I can run tao_byom now.