update
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
@@ -23,6 +23,6 @@ timm
|
|
23 |
omegaconf
|
24 |
hydra-core
|
25 |
wandb
|
26 |
-
torch==2.
|
27 |
-
torchvision==0.
|
28 |
-
torchaudio==2.
|
|
|
23 |
omegaconf
|
24 |
hydra-core
|
25 |
wandb
|
26 |
+
torch==2.6.0 --index-url https://download.pytorch.org/whl/cu124
|
27 |
+
torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
|
28 |
+
torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
|