Added conda environment file for installing with nvidea support using…

… cuda.
SDCCA · Dec 3, 2024 · 3a7bb02 · 3a7bb02
1 parent 0c6bb81
commit 3a7bb02
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/environment_cuda.yaml b/environment_cuda.yaml
@@ -0,0 +1,11 @@
+# To run dgl_ptm on an infra with NVIDIA GPUs, you need to install cuda version
+# of dgl. This environment file specifies the dgl conda channel and cuda version
+# of dgl.
+name: dgl_abm_cuda
+channels:
+  - defaults
+dependencies:
+  - python>=3.11,<3.12
+  - dglteam/label/cu121::dgl
+  - pip:
+    - -e ./dgl_abm  # install dgl_abm in the editable mode, # TODO change to the actual pypi packages