Add nvfuser to requirements.txt

Lightning-AI / lightning-thunder

Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.

Apache License 2.0

1.16k stars 77 forks source link

Add nvfuser to requirements.txt #438

Open vedaanta opened 4 months ago

vedaanta commented 4 months ago

🚀 Feature

Today, nvfuser requirements a manual install. It is NOT part of requirements.txt. Ideally users should be able to get all default executors without having to go through the README.MD and then manually install them.

Motivation

This is a followup task for discussion on #427. LINK to thread

Pitch

Add nvfuser-cu121[torch] to requirements/base.txt.

CC @IvanYashchuk @t-vi

mruberry commented 3 months ago

triage review —

we should consider cpu-only builds (even if they are only for development)
maybe there's an additional .txt file we can provide so people can opt-in to installing nvfuser