Skip to content

Update requirements.txt and use optional dependencies #4

Open
@Anjum48

Description

@Anjum48

It would be good to have an updated requirements.txt since there are several dependencies that are not listed (e.g. onnxoptimizer, nvidia-tensorrt, onnxruntime, transformer-deploy) and also requires TVM to be installed.

If any of these are not core to VoltaML it might be worth adopting an installation method that allows the user to select what is needed, e.g:

pip install voltaml[minimal]
pip install volatml[tvm]
etc...

If TVM is a core requirement, it would probably be worth documenting that since building TVM from source may not be trivial in some environments (e.g. Kaggle GPU notebooks, Colab etc)

Metadata

Metadata

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions