We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.
Already on GitHub? Sign in to your account
I am trying to run this code block:
import torch.nn.functional as F from torch_geometric.nn import CuGraphSAGEConv class CuGraphSAGE(torch.nn.Module): def __init__(self, in_channels, hidden_channels, out_channels, num_layers): super().__init__() self.convs = torch.nn.ModuleList() self.convs.append(CuGraphSAGEConv(in_channels, hidden_channels)) for _ in range(num_layers - 1): conv = CuGraphSAGEConv(hidden_channels, hidden_channels) self.convs.append(conv) self.lin = torch.nn.Linear(hidden_channels, out_channels) def forward(self, x, edge, size): edge_csc = CuGraphSAGEConv.to_csc(edge, (size[0], size[0])) for conv in self.convs: x = conv(x, edge_csc)[: size[1]] x = F.relu(x) x = F.dropout(x, p=0.5) return self.lin(x) model = CuGraphSAGE(128, 64, 349, 3).to(torch.float32).to("cuda") optimizer = torch.optim.Adam(model.parameters(), lr=0.01)
but I get this:
--------------------------------------------------------------------------- ModuleNotFoundError Traceback (most recent call last) Cell In[26], line 25 21 x = F.dropout(x, p=0.5) 23 return self.lin(x) ---> 25 model = CuGraphSAGE(128, 64, 349, 3).to(torch.float32).to("cuda") 26 optimizer = torch.optim.Adam(model.parameters(), lr=0.01) Cell In[26], line 9, in CuGraphSAGE.__init__(self, in_channels, hidden_channels, out_channels, num_layers) 6 super().__init__() 8 self.convs = torch.nn.ModuleList() ----> 9 self.convs.append(CuGraphSAGEConv(in_channels, hidden_channels)) 10 for _ in range(num_layers - 1): 11 conv = CuGraphSAGEConv(hidden_channels, hidden_channels) File ~/.conda/envs/rapids-24.04/lib/python3.11/site-packages/torch_geometric/nn/conv/cugraph/sage_conv.py:40, in CuGraphSAGEConv.__init__(self, in_channels, out_channels, aggr, normalize, root_weight, project, bias) 30 def __init__( 31 self, 32 in_channels: int, (...) 38 bias: bool = True, 39 ): ---> 40 super().__init__() 42 if aggr not in ['mean', 'sum', 'min', 'max']: 43 raise ValueError(f"Aggregation function must be either 'mean', " 44 f"'sum', 'min' or 'max' (got '{aggr}')") File ~/.conda/envs/rapids-24.04/lib/python3.11/site-packages/torch_geometric/nn/conv/cugraph/base.py:41, in CuGraphModule.__init__(self) 38 super().__init__() 40 if not HAS_PYLIBCUGRAPHOPS and not LEGACY_MODE: ---> 41 raise ModuleNotFoundError(f"'{self.__class__.__name__}' requires " 42 f"'pylibcugraphops>=23.02'") ModuleNotFoundError: 'CuGraphSAGEConv' requires 'pylibcugraphops>=23.02'
conda
pip
torch-scatter
The text was updated successfully, but these errors were encountered:
Which version of pylibcugraphops did you have installed?
pylibcugraphops
Sorry, something went wrong.
No branches or pull requests
馃樀 Describe the installation problem
I am trying to run this code block:
but I get this:
Environment
conda
,pip
, source): condatorch-scatter
):The text was updated successfully, but these errors were encountered: