Create environment.yml

inoue0426 · May 28, 2024 · 017e7fc · 017e7fc
1 parent 34422b5
commit 017e7fc
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -0,0 +1,20 @@
+name: drGAT
+channels:
+  - defaults
+  - conda-forge
+dependencies:
+  - python=3.10
+  - numpy==1.23.5
+  - pandas==2.0.3
+  - matplotlib==3.7.1
+  - pip
+  - pip:
+      - optuna==3.2.0
+      - torch==1.13.1+cu116
+      - torch-cluster==1.6.1+pt113cu116
+      - torch-geometric==2.3.1
+      - torch-scatter==2.1.1+pt113cu116
+      - torch-sparse==0.6.17+pt113cu116
+      - torch-spline-conv==1.2.2+pt113cu116
+
+prefix: /path/to/your/envs/drGAT