qitaoz commited on
Commit
6b54bd7
Β·
verified Β·
1 Parent(s): 07c92f0

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -5
requirements.txt CHANGED
@@ -1,13 +1,12 @@
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
 
 
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
+ # ---------------- community wheel feed that hosts PyTorch3D ----------------
7
  --extra-index-url https://miropsota.github.io/torch_packages_builder
8
+ pytorch3d==0.7.7+pt2.2.2cu121 # ← EXACT tag; no git line!
9
 
 
 
10
 
11
  git+https://github.com/facebookresearch/[email protected]#egg=pytorch3d
12