51工具盒子

依楼听风雨
笑看云卷云舒,淡观潮起潮落

Centos7运行python脚本时报错No module named 'onnx_graphsurgeon'

在运行模型python脚本时报错如下内容:

Traceback (most recent call last):
  File "tools/onnx_custom.py", line 4, in <module>
    import onnx_graphsurgeon
ModuleNotFoundError: No module named 'onnx_graphsurgeon'

尝试安装报错:

[root@23cd70ab35ae data1]# python3 -m pip install onnx_graphsurgeon --trusted-host  pypi.tuna.tsinghua.edu.cn -i https://pypi.tuna.tsinghua.edu.cn/simple
WARNING: Running pip install with root privileges is generally not a good idea. Try `python3 -m pip install --user` instead.
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Collecting onnx_graphsurgeon
  Downloading https://pypi.tuna.tsinghua.edu.cn/packages/e8/c8/26f338891498d8a193a9d1698832399979636273f26a67a555e3952a3a7f/onnx-graphsurgeon-0.0.1.dev5.tar.gz
    ERROR: Command errored out with exit status 1:
     command: /usr/bin/python3 -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-i8fljf91/onnx-graphsurgeon/setup.py'"'"'; __file__='"'"'/tmp/pip-install-i8fljf91/onnx-graphsurgeon/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(__file__);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' egg_info --egg-base /tmp/pip-install-i8fljf91/onnx-graphsurgeon/pip-egg-info
         cwd: /tmp/pip-install-i8fljf91/onnx-graphsurgeon/
    Complete output (16 lines):
    Traceback (most recent call last):
      File "<string>", line 1, in <module>
      File "/tmp/pip-install-i8fljf91/onnx-graphsurgeon/setup.py", line 150, in <module>
        raise RuntimeError(open("ERROR.txt", "r").read())
    RuntimeError:
    ###########################################################################################
    The package you are trying to install is only a placeholder project on PyPI.org repository.
    This package is hosted on NVIDIA Python Package Index.

    This package can be installed as:</code></pre>



 
    $ pip install nvidia-pyindex
    $ pip install onnx-graphsurgeon
    ```
    ###########################################################################################

    ----------------------------------------



 
ERROR: Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output.

 

    #### 解决办法



 
python3 -m pip install nvidia-pyindex --trusted-host pypi.tuna.tsinghua.edu.cn -i https://pypi.tuna.tsinghua.edu.cn/simple
python3 -m pip install onnx-graphsurgeon --trusted-host pypi.tuna.tsinghua.edu.cn -i https://pypi.tuna.tsinghua.edu.cn/simple

赞(0)
未经允许不得转载:工具盒子 » Centos7运行python脚本时报错No module named 'onnx_graphsurgeon'