Kunhak Lee
commited on
Commit
·
2e9511d
1
Parent(s):
fea87d0
Add nvidia-nccl-cu12==2.22.3 and triton==3.0.0
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -50,6 +50,7 @@ nvidia-cufft-cu12==11.0.2.54
|
|
| 50 |
nvidia-curand-cu12==10.3.2.106
|
| 51 |
nvidia-cusolver-cu12==11.4.5.107
|
| 52 |
nvidia-cusparse-cu12==12.1.0.106
|
|
|
|
| 53 |
nvidia-nvjitlink-cu12==12.4.99
|
| 54 |
nvidia-nvtx-cu12==12.1.105
|
| 55 |
objprint==0.2.3
|
|
@@ -89,7 +90,7 @@ toolz==0.12.1
|
|
| 89 |
torch==2.2.0
|
| 90 |
tqdm==4.66.2
|
| 91 |
transformers==4.39.2
|
| 92 |
-
triton==
|
| 93 |
typer==0.11.1
|
| 94 |
typing_extensions==4.10.0
|
| 95 |
tzdata==2024.1
|
|
|
|
| 50 |
nvidia-curand-cu12==10.3.2.106
|
| 51 |
nvidia-cusolver-cu12==11.4.5.107
|
| 52 |
nvidia-cusparse-cu12==12.1.0.106
|
| 53 |
+
nvidia-nccl-cu12==2.22.3
|
| 54 |
nvidia-nvjitlink-cu12==12.4.99
|
| 55 |
nvidia-nvtx-cu12==12.1.105
|
| 56 |
objprint==0.2.3
|
|
|
|
| 90 |
torch==2.2.0
|
| 91 |
tqdm==4.66.2
|
| 92 |
transformers==4.39.2
|
| 93 |
+
triton==3.0.0
|
| 94 |
typer==0.11.1
|
| 95 |
typing_extensions==4.10.0
|
| 96 |
tzdata==2024.1
|