如何在没有 CUDA 的 CPU 上 运行 使用 Pytorch 的 GNN 示例?
How to run a GNN example with Pytorch, on a CPU without CUDA?
我正在尝试编写一个 GNN 示例问题,如给定 link 所示:https://towardsdatascience.com/hands-on-graph-neural-networks-with-pytorch-pytorch-geometric-359487e221a8
我使用的是 Macbook Pro 2016 版,没有 Nvidia 显卡!
示例问题是实现 'CUDA' 工具包。我能以某种方式修改我当前笔记本电脑上的代码和 运行 吗?我已经将数据集做得足够小,这样它不需要很高的计算量并且可以 运行 在我的电脑上!
代码中报错的部分如下!
def train():
model.train()
loss_all = 0
for data in train_loader:
data = data.to(device)
optimizer.zero_grad()
output = model(data)
label = data.y.to(device)
loss = crit(output, label)
loss.backward()
loss_all += data.num_graphs * loss.item()
optimizer.step()
return loss_all / len(train_dataset)
device = torch.device('cuda')
model = Net().to(device) # Net = A class inherited from torch.nn.Module
optimizer = torch.optim.Adam(model.parameters(), lr=0.005)
crit = torch.nn.BCELoss()
train_loader = DataLoader(train_dataset, batch_size=batch_size)
for epoch in range(num_epochs):
train()
错误如下
AssertionError: Torch not compiled with CUDA enabled
您正在使用:
device = torch.device('cuda')
如果您喜欢使用cpu请改为:
device = torch.device('cpu')
我正在尝试编写一个 GNN 示例问题,如给定 link 所示:https://towardsdatascience.com/hands-on-graph-neural-networks-with-pytorch-pytorch-geometric-359487e221a8
我使用的是 Macbook Pro 2016 版,没有 Nvidia 显卡!
示例问题是实现 'CUDA' 工具包。我能以某种方式修改我当前笔记本电脑上的代码和 运行 吗?我已经将数据集做得足够小,这样它不需要很高的计算量并且可以 运行 在我的电脑上!
代码中报错的部分如下!
def train():
model.train()
loss_all = 0
for data in train_loader:
data = data.to(device)
optimizer.zero_grad()
output = model(data)
label = data.y.to(device)
loss = crit(output, label)
loss.backward()
loss_all += data.num_graphs * loss.item()
optimizer.step()
return loss_all / len(train_dataset)
device = torch.device('cuda')
model = Net().to(device) # Net = A class inherited from torch.nn.Module
optimizer = torch.optim.Adam(model.parameters(), lr=0.005)
crit = torch.nn.BCELoss()
train_loader = DataLoader(train_dataset, batch_size=batch_size)
for epoch in range(num_epochs):
train()
错误如下
AssertionError: Torch not compiled with CUDA enabled
您正在使用:
device = torch.device('cuda')
如果您喜欢使用cpu请改为:
device = torch.device('cpu')