🚑 fix cuda deps, prevent future me from dying of unreproducibility

environment.yml

@@ -1,28 +1,37 @@
 name: sap
+
 channels:
-  - pytorch3d
+  - nodefaults
   - conda-forge
   - pytorch
-  - defaults
+  - nvidia
+  - pyg
+
 dependencies:
-  - python
-  - pytorch
-  - torchvision
-  - cudatoolkit
-  - numpy
-  - matplotlib
-  - pyyaml
-  - scipy
+#---# basic python
+  - python=3.8
   - tqdm
+  - pyyaml
+#---# visu
+  - matplotlib
+#---# scientific
+  - numpy
+  - scipy
   - trimesh
   - igl
-  - tensorboard
-  - pytorch3d
+#---# pytorch
+  - pytorch
+  - pytorch-cuda=11.8
+  - cudatoolkit
+  - torchvision
+  - torch-scatter
 #---# tooling (linting, typing...)
   - ruff
   - mypy
   - black
   - isort
+#---# logging
+  - tensorboard
 #---# pip shit
   - pip
   - pip: