调用` `cublasCreate(handle)`时运行pytorch geometric CUDA错误: CUBLAS_STATUS_NOT_INITIALIZED

Question

我正在运行这个笔记本中的代码。https://colab.research.google.com/github/zaidalyafeai/Notebooks/blob/master/Deep_GCN_Spam.ipynb#scrollTo=UjoTbUQVnCz8当我将数据集更改为我自己的数据集时，我得到一个错误。我知道这可能是我的代码错误。然后，我清除了生成数据集的所有代码。我将两个数据集保存为文件并重新加载。我真的看不出这两个数据集之间的区别。

下面提供了这两个数据集的形状和类型。我可以提供所需的任何信息。有人能帮我解决这个问题吗？

这是我的数据集

data = torch.load("dataset.pt")
data
>>>Data(edge_attr=[3585, 1], edge_index=[2, 3585], x=[352, 1], y=[352])
data.x.dtype, data.y.dtype, data.edge_attr.dtype, data.edge_index.dtype
>>>(torch.float32, torch.int64, torch.float32, torch.int64)
data.edge_index.T.numpy().shape
>>>(3585, 2)
np.unique(data.edge_index.T.numpy(), axis=0).shape
>>>(3585, 2)
np.unique(data.edge_index.T.numpy(), axis=0).shape
>>>(3585, 2)
data.edge_index.unique().shape
>>>torch.Size([352])
data.edge_index
>>>tensor([[ 13,  13,  13,  ..., 103, 103, 103],
           [  1,   2,   3,  ...,   6,   9,  10]])

这是笔记本中提到的数据集

data2 = torch.load("spam.pt")
data2
>>>Data(edge_attr=[50344, 1], edge_index=[2, 50344], x=[1000, 1], y=[1000])
data2.x.dtype, data2.y.dtype, data2.edge_attr.dtype, data2.edge_index.dtype
>>>(torch.float32, torch.int64, torch.float32, torch.int64)
data2.edge_index
>>>tensor([[  0,   1,   1,  ..., 999, 999, 999],
           [455, 173, 681,  ..., 377, 934, 953]])

Python版本: 3.8

PyTorch几何版本: 1.6.2

CUDA版本: 10.2

系统: Windows 10

Screeshot

Answer 1

在我的例子中，我需要确保我的edge_attributes在0,1的范围内。

请参阅here