Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
@@ -2,7 +2,7 @@ torch
|
|
2 |
torchvision
|
3 |
cuda-python==11.8
|
4 |
nvidia-pyindex
|
5 |
-
nvidia-
|
6 |
pandas
|
7 |
tiktoken
|
8 |
tqdm
|
|
|
2 |
torchvision
|
3 |
cuda-python==11.8
|
4 |
nvidia-pyindex
|
5 |
+
nvidia-tensorflow[horovod]
|
6 |
pandas
|
7 |
tiktoken
|
8 |
tqdm
|