Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
pytorch==2.1.2
|
| 3 |
torchvision==0.16.2
|
| 4 |
torchaudio==2.1.2
|
| 5 |
-
pytorch-cuda
|
| 6 |
termcolor==2.4.0
|
| 7 |
streamlit==1.39.0
|
| 8 |
class_registry==2.1.2
|
|
|
|
| 2 |
pytorch==2.1.2
|
| 3 |
torchvision==0.16.2
|
| 4 |
torchaudio==2.1.2
|
| 5 |
+
pytorch-cuda==12.1
|
| 6 |
termcolor==2.4.0
|
| 7 |
streamlit==1.39.0
|
| 8 |
class_registry==2.1.2
|