Skip to content

Commit 3a7bb02

Browse files
author
thijsvl
committed
Added conda environment file for installing with nvidea support using cuda.
1 parent 0c6bb81 commit 3a7bb02

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

environment_cuda.yaml

+11
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)