I import graphvite with pytorch but it shows a segmentation fault, here's the sample code:
import graphvite as gv
import torch
DEVICE = torch.device("cuda")
torch.rand((2, 3)).to(DEVICE)
And I run it in gdb, it tells me:
Thread 1 "python" received signal SIGSEGV, Segmentation fault.
0x00007fffcd1a77d8 in ?? ()
from /home/wanghan/anaconda3/envs/dane/lib/python3.7/site-packages/graphvite/lib/../../../../libcudart.so.10.1
I would like to know how to fix this? Thanks!