search

mojave-pku / TextLevelGCN

source code of our paper presents in EMNLP 2019. https://www.aclweb.org/anthology/D19-1345/
GNU General Public License v3.0
54 stars 15 forks source link

version of cuda and dgl #6

Open julycetc opened 4 years ago

julycetc commented 4 years ago

Hello, when i run the code,i get the following error: Check failed: allow_missing: Device API gpu is not enabled. Please install the cuda version of dgl. so i wuold like to know what is your version of cuda and dgl? Thanks very much.

fodothing commented 2 years ago

Maybe this code could help you, good luck!

import torch
try:
  import dgl
except ModuleNotFoundError:
  CUDA = 'cu' + torch.version.cuda.replace('.','')
  !pip install dgl-{CUDA} -f https://data.dgl.ai/wheels/repo.html