Сбой сеанса Colab при импорте torch_geometric.data

Question

Моя сессия colab всегда падает при попытке импортировать модуль torch_geometric.data. Для справки, код, который я пишу, выглядит следующим образом:

      import torch

def format_pytorch_version(version):

  return version.split('+')[0]

TORCH_version = torch.__version__

TORCH = format_pytorch_version(TORCH_version)

def format_cuda_version(version):

  return 'cu' + version.replace('.', '')

CUDA_version = torch.version.cuda

CUDA = format_cuda_version(CUDA_version)

!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

!pip install torch_geometric

from torch_geometric.data import Data 

Когда выполнение достигает линии lats, сессия colab падает, я пробовал также переключаться на GPU и TPU, но ничего не происходит.

Логи показывают это:

ВНИМАНИЕ:root: ядро ​​​​fc8fda2d-f1ce-4808-b78f-a18a55132346 перезапущено

Answer 1

На этой неделе возникла такая же проблема.Понижение PyTorch до той же версии, что и pytorch_geometric, похоже, решило проблему.

      !pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html