Spaces:
Running
on
T4
Running
on
T4
Update requirements.txt
Browse files- requirements.txt +7 -2
requirements.txt
CHANGED
@@ -1,6 +1,11 @@
|
|
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
2 |
-
torch==2.2.2+cu121
|
3 |
-
torchvision==0.17.2+cu121
|
|
|
|
|
|
|
|
|
4 |
|
5 |
git+https://github.com/facebookresearch/[email protected]#egg=pytorch3d
|
6 |
|
|
|
1 |
+
# βββββββ 1. official PyTorch wheels (CUDA 12.1 runtime) βββββββ
|
2 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
3 |
+
torch==2.2.2+cu121
|
4 |
+
torchvision==0.17.2+cu121
|
5 |
+
|
6 |
+
# βββββββ 2. community wheel feed that hosts PyTorch3D for cu121 βββββββ
|
7 |
+
--extra-index-url https://miropsota.github.io/torch_packages_builder
|
8 |
+
pytorch3d==0.7.7+pt2.2.2cu121 # pre-built wheel β no compile, no torch import
|
9 |
|
10 |
git+https://github.com/facebookresearch/[email protected]#egg=pytorch3d
|
11 |
|