search

NVIDIA / tao_pytorch_backend

TAO Toolkit deep learning networks with PyTorch backend
Apache License 2.0
85 stars 18 forks source link

ModuleNotFoundError: No module named 'nvidia_tao_core' #26

Open SInnget opened 2 months ago

SInnget commented 2 months ago

Describe the bug

in the branch tao_5.5_release, we try to import package from nvidia_tao_core package

e.g. https://github.com/NVIDIA/tao_pytorch_backend/blob/dc07b02eb78c2eb868315107892b466496e55a0f/nvidia_tao_pytorch/core/entrypoint.py#L31

but after building the docker image, we find not module name nvidia_tao_core in side. which may related to #25

dharagg07 commented 2 months ago

Did you ever got a fix for it?

morganh-nv commented 1 month ago

Please modify to below. from nvidia_tao_pytorch.core.telemetry.telemetry import send_telemetry_data

SInnget commented 1 month ago

Please modify to below. from nvidia_tao_pytorch.core.telemetry.telemetry import send_telemetry_data

there are three parts which import "nvidia_tao_core", some of them does not find any alternative functions

morganh-nv commented 1 month ago

As a workaround, please directly pull the docker nvcr.io/nvidia/tao/tao-toolkit:5.5.0-pyt and run inside it. And then you can find any code and modify any code.