We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0c6bb81 commit 3a7bb02Copy full SHA for 3a7bb02
environment_cuda.yaml
@@ -0,0 +1,11 @@
1
+# To run dgl_ptm on an infra with NVIDIA GPUs, you need to install cuda version
2
+# of dgl. This environment file specifies the dgl conda channel and cuda version
3
+# of dgl.
4
+name: dgl_abm_cuda
5
+channels:
6
+ - defaults
7
+dependencies:
8
+ - python>=3.11,<3.12
9
+ - dglteam/label/cu121::dgl
10
+ - pip:
11
+ - -e ./dgl_abm # install dgl_abm in the editable mode, # TODO change to the actual pypi packages
0 commit comments