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

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -4
requirements.txt CHANGED
@@ -1,11 +1,13 @@
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
 
 
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