Spaces:
Running
on
T4
Running
on
T4
Update requirements.txt
Browse files- requirements.txt +6 -4
requirements.txt
CHANGED
@@ -1,11 +1,13 @@
|
|
1 |
-
#
|
2 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
3 |
torch==2.2.2+cu121
|
4 |
-
torchvision==0.17.2+cu121
|
5 |
|
6 |
-
#
|
7 |
--extra-index-url https://miropsota.github.io/torch_packages_builder
|
8 |
-
|
|
|
|
|
9 |
|
10 |
git+https://github.com/facebookresearch/[email protected]#egg=pytorch3d
|
11 |
|
|
|
1 |
+
# 1 β PyTorch & friends (official CUDA 12.1 wheels)
|
2 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
3 |
torch==2.2.2+cu121
|
4 |
+
torchvision==0.17.2+cu121 # must match the Torch build
|
5 |
|
6 |
+
# 2 β Extra index that stores the PyTorch3D wheels
|
7 |
--extra-index-url https://miropsota.github.io/torch_packages_builder
|
8 |
+
|
9 |
+
# 3 β Pre-built PyTorch3D wheel β NO git line!
|
10 |
+
pytorch3d==0.7.7+pt2.2.2cu121
|
11 |
|
12 |
git+https://github.com/facebookresearch/[email protected]#egg=pytorch3d
|
13 |
|