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

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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 # any 2.x wheel with β€œ+cu121” works on the 12.3 image
3
- torchvision==0.17.2+cu121 # keep the matching torchvision
 
 
 
 
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