search

airockchip / rknn-toolkit2

Other
806 stars 83 forks source link

Missing pyyaml module #139

Open cadeniyi opened 3 weeks ago

cadeniyi commented 3 weeks ago

Using the v2.1.0 tag I get this error:

ModuleNotFoundError: No module named 'yaml'

when trying to run the rknn_convert script within the dockerised setup.

Installing the pyyaml module using pip3 (within the running container) seems to fix this.

My suggestion for fixing is to add:

RUN pip3 install pyyaml

to the Dockerfile

zen-xingle commented 2 weeks ago

In the requirement file, "ruamel.yaml" is declared and it's shouldn't missing the 'yaml' module. If more user report this error, maybe the better solution is change "ruamel.yaml" to "pyyaml" requirements.

Anyway, Thank you for reporting this bug.