Update requirements.txt

requirements.txt

@@ -1,6 +1,8 @@
-torch==2.0.1
-torchvision==0.15.2
-pytorch3d==0.7.4
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.2.2+cu121          # any 2.x wheel with “+cu121” works on the 12.3 image
+torchvision==0.17.2+cu121   # keep the matching torchvision
+
+git+https://github.com/facebookresearch/[email protected]#egg=pytorch3d
 
 black
 einops