TextLevelGCN icon indicating copy to clipboard operation
TextLevelGCN copied to clipboard

version of cuda and dgl

Open julycetc opened this issue 4 years ago • 1 comments

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.

julycetc avatar Jun 11 '20 09:06 julycetc

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

fodothing avatar Feb 10 '22 08:02 fodothing