TextLevelGCN
TextLevelGCN copied to clipboard
version of cuda and dgl
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.
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