Commit ef042d4d authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Fixing cpu dependencies

parent 45be6965
Loading
Loading
Loading
Loading
+6 −8
Original line number Diff line number Diff line
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==1.5.9
    - torch-scatter==2.0.6
    - torch-sparse==0.6.9
    - torch-spline-conv==1.2.1
    - -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html
    - dgl==0.6.*
    - torch==1.8.0+cpu
    - torchvision==0.9.0+cpu
    - torch-scatter
    - torch-sparse