gpt4 book ai didi

python - 没有名为 'uff' 的模块

转载 作者:行者123 更新时间:2023-12-02 19:38:39 25 4
gpt4 key购买 nike

目标: 将tensorflow .pb模型转换为tensorrt

系统规范:乌本杜 18.04库达10.0TensorRT 5.1

甚至已安装sudo apt-get install uff-converter-tf

尝试在 Python 中导入 uff 时出现错误:

ModuleNotFoundError: No module named 'uff'

最佳答案

正在安装uff-converter-tf还不够,您需要安装Python UFF Wheel才能使用它。 Here您将找到完整的安装指南。

如果使用 Python 2.7:

pip2 install tensorrt-*-cp27-none-linux_x86_64.whl

如果使用 Python 3.x:

pip3 install tensorrt-*-cp3x-none-linux_x86_64.whl

您会发现这些轮子以及解压后生成的 TensorRT 二进制文件的其余部分 TensorRT-${version}.${os}.${arch}-gnu.${cuda}.${cudnn}.tar.gz

一如既往,最好安装在虚拟环境中。

关于python - 没有名为 'uff' 的模块,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60653617/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com