Pin torch/torchvision/torchaudio to 2.1.0+cu121: unpinned torch was drifting to latest PyPI (now 2.14.0/cu13x), breaking ABI with the hard-pinned kaolin/pytorch3d/torch_scatter cu121 wheels (libcudart.so.13 not found)
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
spaces
|
| 2 |
setuptools<82
|
| 3 |
-
torch
|
| 4 |
-
torchvision
|
| 5 |
-
torchaudio --index-url https://download.pytorch.org/whl/cu121
|
| 6 |
huggingface_hub==0.25.2
|
| 7 |
absl-py==2.1.0
|
| 8 |
accelerate==1.1.1
|
|
|
|
| 1 |
spaces
|
| 2 |
setuptools<82
|
| 3 |
+
torch==2.1.0 --index-url https://download.pytorch.org/whl/cu121
|
| 4 |
+
torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu121
|
| 5 |
+
torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
|
| 6 |
huggingface_hub==0.25.2
|
| 7 |
absl-py==2.1.0
|
| 8 |
accelerate==1.1.1
|