env.cpu.yml

dependencies:
  - pip:
    - -f https://download.pytorch.org/whl/torch_stable.html
    - -f https://pytorch-geometric.com/whl/torch-1.6.0+cpu.html
    - dgl==0.5.*
    - torch==1.6.0+cpu
    - torchvision==0.7.0+cpu
    - torch-cluster
    - torch-scatter
    - torch-sparse
    - torch-spline-conv