ONNX PyTorch
| Name: | flytekitplugins-onnxpytorch |
| Version: | 0.0.0+develop |
| Author: | admin@flyte.org |
| Provides: |
flytekitplugins.onnxpytorch |
| Requires: |
flytekit>=1.3.0b2,<2.0.0 torch>=1.11.0 onnx-pytorch networkx<3.2; python_version<'3.9' |
| Python: | >=3.9 |
| License: | apache2 |
| Source Code: | https://github.com/flyteorg/flytekit/tree/master/plugins/flytekit-onnx-pytorch |
- Intended Audience :: Science/Research
- Intended Audience :: Developers
- License :: OSI Approved :: Apache Software License
- Programming Language :: Python :: 3.9
- Programming Language :: Python :: 3.10
- Topic :: Scientific/Engineering
- Topic :: Scientific/Engineering :: Artificial Intelligence
- Topic :: Software Development
- Topic :: Software Development :: Libraries
- Topic :: Software Development :: Libraries :: Python Modules
This plugin allows you to generate ONNX models from your PyTorch models.
To install the plugin, run the following command:
pip install flytekitplugins-onnxpytorch