Skip to content

Commit ad8ed75

Browse files
authored
move to nvfuser-cu128-torch28 (#2604)
1 parent 8b542cf commit ad8ed75

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

examples/coverage/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
transformers==4.52.4
22
accelerate
3-
nvfuser-cu128-torch27
3+
nvfuser-cu128-torch28
44
nvidia-cudnn-frontend
55
einops
66
tiktoken
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
transformers==4.55.4
22
accelerate
3-
nvfuser-cu128-torch27
3+
nvfuser-cu128-torch28
44
nvidia-cudnn-frontend

thunder/recipes/base.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,10 +34,11 @@ def get_nvfuser_package_hint() -> str:
3434
else:
3535
return f"""nvFuser was specified but we don't currently support torch {torch_version}.
3636
37-
Please upgrade to torch 2.6 or 2.7 and then run
37+
Please upgrade to torch 2.6 to 2.8 and then run
3838
```
3939
pip install nvfuser-cu126-torch26 # for torch 2.6
4040
pip install nvfuser-cu128-torch27 # for torch 2.7
41+
pip install nvfuser-cu128-torch28 # for torch 2.8
4142
```
4243
4344
For compatibility options, see:

0 commit comments

Comments
 (0)