Program halts whenever trying to import ogb.graphproppred

Hi I am able to run from torch_geometric.data import InMemoryDataset successfully. But whenever I try to import anything from ogb, the program stuck for infinite time. Import code is: from ogb.graphproppred import Evaluator

I followed the following command to install torch in conda environment:

CUDA=cu113 TORCH=1.11.0 pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html pip install torch-geometric Note that, I am using Ubuntu. My Nvidia-smi output:

$ nvidia-smi Wed Apr 20 05:09:04 2022
±----------------------------------------------------------------------------+ | NVIDIA-SMI 450.119.03 Driver Version: 450.119.03 CUDA Version: 11.0 | |-------------------------------±---------------------±---------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |===============================+======================+======================| | 0 GeForce GTX 1070 Off | 00000000:01:00.0 On | N/A | | 27% 36C P8 9W / 151W | 1227MiB / 8116MiB | 1% Default | | | | N/A

[faysalhossain2007]

Thanks for the help! At last I am able to run the code with small dataset with a few modification to the code.

[faysalhossain2007]

Thought of other might get help so posting here my detailed approach:

conda create --name=pyg9 python=3.9
conda activate pyg9
conda install pyg -c pyg
conda install torchvision
conda install ogb

In main.py => I move from ogb.graphproppred import Evaluator to the beginning of the file

In src/data/dataloader.py =>

import torch
from torch.utils.data.dataloader import default_collate
from torch_geometric.data import Data, Batch
from torch._six import  string_classes
import collections.abc as container_abcs

In Line 44: I applied the following modifications:

elif isinstance(elem, int_classes): changed to elif isinstance(elem, int):