Lightning-AI / Lightning-AI/lightning-thunder

ModuleNotFoundError: No module named 'extensions'

Open
#2,269 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

bug
Dominant language
Python
Stars
1.5k
Forks
121
PR merge metrics
No merged PRs in 30d

Description

### Bug description

Unable to run with example code in https://github.com/Lightning-AI/litgpt/blob/main/extensions/thunder/README.md:
`
python extensions/thunder/pretrain.py --config config.yaml --executors '[sdpa, torchcompile_cat, nvfuser, torch]'
`

### What operating system are you using?

Linux

### LitGPT Version

```
Version: 0.4.3.dev0
```

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the command in extensions/thunder/README.md and the entry point extensions/thunder/pretrain.py, then reproduce the ModuleNotFoundError on Linux using the reported version. Trace how the example is launched and verify that the documented command runs without the missing-module error.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
compilers
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
38/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.