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

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

5 голосов
/ 15 февраля 2020

Я работал над проектом PyTorch Geometri c, используя Google Colab для поддержки CUDA. Поскольку ее библиотеки по умолчанию нет, я запускаю: 

!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

Недавно, при импорте torch_geometri c, из-за обновлений версий, существует несоответствие версии CUDA:

RuntimeError: Обнаружено, что PyTorch и torch_sparse были скомпилированы с различными версиями CUDA. PyTorch имеет версию 10.1 CUDA, а torch_sparse имеет версию 10.0 CUDA. Пожалуйста, переустановите torch_sparse, который соответствует вашей установке PyTorch.

Чтобы решить эту проблему, я попытался использовать conda для спецификации c CUDA версии как:

!conda install pytorch==1.4.0 cudatoolkit=10.0 -c pytorch

Тем не менее, при запуске print(torch.version.cuda) я получаю 10.1 в качестве вывода, а не 10.0, как я хотел.

Это недавняя ошибка, так как она не поднимала эту проблему на прошлой неделе. Любая лучшая практика, чтобы решить эту проблему?

Ответы [ 3 ]

4 голосов
/ 15 февраля 2020

С их веб-сайт

Попробуйте это 

!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
0 голосов
/ 01 апреля 2020

вы можете попробовать следующее, чтобы увидеть, решит ли это вашу проблему с ошибкой управления версиями CUDA в "pytorch-geometryri c":

  1. apt-get --purge remove " cublas"" cuda * "
  2. перезагрузка
  3. sudo curl -O http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
  4. sudo dpkg -i cuda-repo -ubuntu1804_10.0.130-1_amd64.deb sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
  5. sudo apt-get установить cuda-10-1
  6. python - c "импортный факел; печать (torch.version.cuda)"

    10.1

  7. nv cc - версия

    Инструменты компиляции Cuda, выпуск 10.1, V10.1.243

0 голосов
/ 07 марта 2020

Проблемы можно решить с помощью комментария: 

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

Есть ли у нас другое решение?

...