Как запустить пример GNN с Pytorch на процессоре без CUDA?

Question

Я пытаюсь закодировать пример проблемы GNN, как показано в данной ссылке: 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

Answer 1

Ты используешь:

      device = torch.device('cuda')

Если вы хотите использовать процессор, измените его на:

      device = torch.device('cpu')