From 377132fdd59f9f94b60afbb91968bb8243798576 Mon Sep 17 00:00:00 2001
From: Laurent FAINSIN <laurent.fainsin@safrangroup.com>
Date: Tue, 30 May 2023 11:45:36 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=91=20fix=20cuda=20deps,=20prevent=20f?=
 =?UTF-8?q?uture=20me=20from=20dying=20of=20unreproducibility?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 environment.yml | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/environment.yml b/environment.yml
index 4b0211b..0911f67 100644
--- a/environment.yml
+++ b/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: