Add environment.yml

eigenvivek · Nov 25, 2023 · c1bcd60 · c1bcd60
1 parent 65a45a7
commit c1bcd60
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -0,0 +1,22 @@
+name: preop
+channels:
+  - conda-forge
+  - pytorch
+  - pytorch3d
+  - nvidia
+dependencies:
+  - pip
+  - pytorch
+  - torchvision
+  - pytorch3d
+  - pip:
+    - diffdrr
+    - h5py
+    - scikit-image
+    - seaborn
+    - transformers
+    - timm
+    - torchmetrics
+    - tqdm
+    - beartype
+    - jaxtyping