геометрия pytorch c "Обнаружено, что PyTorch и torch_sparse были скомпилированы с разными версиями CUDA" на Google Colab - PullRequest
Новая
       26

геометрия pytorch c "Обнаружено, что PyTorch и torch_sparse были скомпилированы с разными версиями CUDA" на Google Colab

2 голосов
/ 19 апреля 2020

Я новичок в pytorch geometryri c, пытался установить его на свой компьютер, но не смог, поэтому вместо этого я пытаюсь запустить код в Google Colab. Согласно этому предыдущему вопросу (который мне не помог, и я не уверен, что это та же проблема):

PyTorch Geometri c Проблемы установки CUDA в Google Colab

Я сделал: 

!pip install --upgrade torch-scatter

!pip install --upgrade torch-sparse

!pip install --upgrade torch-cluster

!pip install --upgrade torch-spline-conv 

!pip install torch-geometric

!pip install torch-cluster==latest+cu101 -f https://s3.eu-central-1.amazonaws.com/pytorch-geometric.com/whl/torch-1.4.0.html 

!pip install torch-scatter==latest+cu101 torch-sparse==latest+cu101 torch-spline-conv==latest+cu101 -f https://s3.eu-central-1.amazonaws.com/pytorch-geometric.com/whl/torch-1.4.0.html

они печатают: 

Successfully installed torch-cluster-1.5.4

Successfully installed torch-scatter-2.0.4 torch-sparse-0.6.1 torch-spline-conv-1.2.0

Однако, когда я пытаюсь запустить 

import torch_geometric.datasets as datasets

, я получаю: 

RuntimeError: Detected that PyTorch and torch_sparse were compiled with different CUDA versions. PyTorch has CUDA version 10.1 and torch_sparse has CUDA version 0.0. Please reinstall the torch_sparse that matches your PyTorch install.

Любая помощь будет высоко оценена.

1 Ответ

0 голосов
/ 19 апреля 2020

Я обновил ответ на этот предыдущий вопрос. Надеюсь, это работает сейчас. Также не забудьте выбрать GPU Runtime. 

!pip install torch-geometric \
  torch-sparse==latest+cu101 \
  torch-scatter==latest+cu101 \
  torch-cluster==latest+cu101 \
  -f https://pytorch-geometric.com/whl/torch-1.4.0.html
...