Unverified Commit df00e5f4 authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

Update env.gpu.yml

parent d0491e9b

env.gpu.yml

+1 −1
Original line number Diff line number Diff line
@@ -2,7 +2,7 @@ dependencies: 
  - pip:

    - -f https://download.pytorch.org/whl/torch_stable.html

    - -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html

    - dgl-cu101==0.5.*

    - dgl-cu101==0.6.*

    - torch==1.6.0+cu101

    - torchvision==0.7.0+cu101

    - torch-cluster

CRA Git | Maintained and supported by SUSTech CRA and CCSE