Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update requirements.txt and use optional dependencies #4

Open
Anjum48 opened this issue Oct 31, 2022 · 1 comment
Open

Update requirements.txt and use optional dependencies #4

Anjum48 opened this issue Oct 31, 2022 · 1 comment
Assignees

Comments

@Anjum48
Copy link

Anjum48 commented Oct 31, 2022

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)

@riteshgangnani10 riteshgangnani10 self-assigned this Nov 1, 2022
@VoltaML
Copy link
Collaborator

VoltaML commented Nov 2, 2022

Hi @Anjum48. Thanks for your suggestion. We will update our requirements soon and also have installation methods for different compilers.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants