RuntimeError with GCN predictor in installation test about naszilla HOT 2 OPEN

A quick adjacent question: multi-GPU acceleration is currently not considered by the implementation, correct?

from naszilla.

Hi, with the release of nas-bench-301, there is no place in this repository where a GPU is needed. All of the experiments were run on single CPUs, and I would recommend using a CPU.

If you still want to use a GPU instead of a CPU, here is something you can try:
(1) Verify that this line is printing cuda:0, or replace it with device = torch.device("cuda:0").
(2) Verify that adj.device is set to cuda:0 here, or else replace that line with adj_with_diag = normalize_adj(adj + torch.eye(gs, device=torch.device("cuda:0")))

I think that will work, but it will be a few days before I can check this on a GPU.

from naszilla.