dgl_dev.yml.template 530 Bytes
Newer Older
1
2
name: __NAME__
dependencies:
3
  - python=__PYTHON_VERSION__
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
  - pip
  - pip:
    - --find-links https://download.pytorch.org/whl/torch_stable.html
    - cython
    - filelock
    - matplotlib
    - networkx
    - nltk
    - nose
    - numpy
    - ogb
    - pandas
    - psutil
    - pyarrow
    - pydantic
    - pytest
    - pyyaml
    - rdflib
    - requests[security]
    - scikit-learn
    - scipy
    - torch==__TORCH_VERSION__
    - torchmetrics
    - tqdm
28
    - boto3 # AWS SDK for python
29
30
variables:
  DGL_HOME: __DGL_HOME__