I am having issues making DGL work with CUDA, this is my current setup:
CUDA Version: 10.2
dgl==0.4.3.post2
dgl-cu102==0.4.3.post2
Cuda works correctly (I use it with both Keras and Torch):
>>> torch.cuda.is_available()
True
>>> torch.device('cuda:0')
device(type='cuda', index=0)
>>> torch.zeros(1).cuda()
tensor([0.], device='cuda:0')
But of course when I run:
>>> g.to(th.device(device)
I get:
DGLError: [17:03:22] /opt/dgl/src/runtime/c_runtime_api.cc:87: Check failed: allow_missing: Device API gpu is not enabled. Please install the cuda version of dgl.
So I tried to install and compile DGL from sources (enabling CUDA with CMAKE) but the result doesn’t change. What am I missing?
Thank you!