如何解决正在运行的pytorch几何获取CUDA错误:调用`cublasCreatehandle`
我正在运行此笔记本中的代码。 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,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)
data.edge_index.unique().shape
>>>torch.Size([352])
data.edge_index
>>>tensor([[ 13,13,...,103,103],[ 1,2,3,6,9,10]])
这是笔记本中提到的数据集
data2 = torch.load("spam.pt")
data2
>>>Data(edge_attr=[50344,50344],x=[1000,y=[1000])
data2.x.dtype,data2.y.dtype,data2.edge_attr.dtype,data2.edge_index.dtype
>>>(torch.float32,torch.int64)
data2.edge_index
>>>tensor([[ 0,1,999,999],[455,173,681,377,934,953]])
Python版本:3.8
PyTorch几何版本:1.6.2
CUDA版本:10.2
系统:Windows 10
解决方法
对于我来说,我需要确保我的edge_attributes在[0,1]范围内。
请参见here
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。