TensorRT
TensorRT copied to clipboard
Move `onnx-graphsurgeon` into a separate repo
Description
onnx-graphsurgeon is an extremily helpful tool used as a dependency in multiple other tools But it is unfortunately not available in pypi. Also I prefer to install pure python deps from source, and it is inconvenient to download whole TensorRT just to get onnx-graphsurgeon.
It is proposed to:
- extract
graphsurgeoninto a separate git repo. - publish it on
pypi. - setup CI publishing wheels to
pypi
P.S. pypi has a package with such a name, but it looks like a stub left by NVidia to reserve the name.
You can install it from the NVIDIA PyPI as mentioned in the README:
python3 -m pip install onnx_graphsurgeon --index-url https://pypi.ngc.nvidia.com
You can install it from the NVIDIA PyPI as mentioned in the README:
I know, but I prefer to install from git.
Also I'm not sure that PEP 508 allows to specify --index-url to use it in package dependencies (not requirements.txt, but PEP 621-compliant pyproject.toml).
It surely allows to specify whole this repo (with a subdir within it), but it will result in fetching the whole repo, since neither pip, nor GitHub support partial fetches.
Does a URL to the wheel work? https://developer.download.nvidia.com/compute/redist/onnx-graphsurgeon/onnx_graphsurgeon-0.3.20-py2.py3-none-any.whl
It should work, but pinning to an exact version is highly undesireable.
I guess you can try
python3 -m pip install -U 'git+https://github.com/NVIDIA/TensorRT.git#subdirectory=tools/onnx-graphsurgeon'
But second on move onnx-graphsurgeon into a separate repo.
closing since there is already solution, thanks all!