Spaces:
Running
on
L4
Running
on
L4
Updated PyTorch to 2.1.2 with CUDA 12.1 index and enabled faiss-gpu. Pinned torch/torchvision/torchaudio versions to prevent mismatches.
Browse files- requirements.txt +6 -2
requirements.txt
CHANGED
|
@@ -7,8 +7,12 @@ pyarrow
|
|
| 7 |
shapely
|
| 8 |
pandas
|
| 9 |
numpy
|
| 10 |
-
|
| 11 |
-
torch
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
tqdm
|
| 13 |
requests
|
| 14 |
aiohttp
|
|
|
|
| 7 |
shapely
|
| 8 |
pandas
|
| 9 |
numpy
|
| 10 |
+
|
| 11 |
+
torch==2.1.2
|
| 12 |
+
torchvision==0.16.2
|
| 13 |
+
torchaudio==2.1.2
|
| 14 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 15 |
+
|
| 16 |
tqdm
|
| 17 |
requests
|
| 18 |
aiohttp
|