23

Google Colab: PyTorch Geometric Installation

 1 year ago
source link: https://gist.github.com/ameya98/b193856171d11d37ada46458f60e73e7
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

PyTorch Geometric Installation · GitHub

Instantly share code, notes, and snippets.

Google Colab: PyTorch Geometric Installation

Hi, I've tried your code, but it still failed to import torch_geometric. Have you ever faced this issue?

Author

ameya98 commented on Oct 4, 2020

Ah, I expected this to happen eventually. This script is quite rudimentary. Can you please print out the variables TORCH_version and CUDA_version in the script above?

Author

ameya98 commented on Oct 4, 2020

edited

I just tested this on the CPU, GPU and TPU runtimes on Colab, and I can successfully import PyG.

This worked for me - thank you!

Hi,
For some reason this code worked perfectly well once and dosn't work any more now.
Do you know about any changes in Colab/PyTorch between the 26 ad 27 February ?
I get the following error:

Looking in links: https://pytorch-geometric.com/whl/torch-1.7.1.html
ERROR: Could not find a version that satisfies the requirement torch-scatter==latest+cu101 (from versions: 0.3.0, 1.0.2, 1.0.3, 1.0.4, 1.1.0, 1.1.1, 1.1.2, 1.2.0, 1.3.0, 1.3.1, 1.3.2, 1.4.0, 2.0.2, 2.0.3, 2.0.4, 2.0.5)
ERROR: No matching distribution found for torch-scatter==latest+cu101

Looking in links: https://pytorch-geometric.com/whl/torch-1.7.1.html
ERROR: Could not find a version that satisfies the requirement torch-sparse==latest+cu101 (from versions: 0.1.0, 0.2.0, 0.2.1, 0.2.2, 0.2.3, 0.2.4, 0.4.0, 0.4.1, 0.4.2, 0.4.3, 0.4.4, 0.5.1, 0.6.0, 0.6.1, 0.6.3, 0.6.4, 0.6.5, 0.6.6, 0.6.7, 0.6.8)
ERROR: No matching distribution found for torch-sparse==latest+cu101

Looking in links: https://pytorch-geometric.com/whl/torch-1.7.1.html
ERROR: Could not find a version that satisfies the requirement torch-cluster==latest+cu101 (from versions: 0.1.1, 0.2.3, 0.2.4, 1.0.1, 1.0.3, 1.1.1, 1.1.2, 1.1.3, 1.1.4, 1.1.5, 1.2.1, 1.2.2, 1.2.3, 1.2.4, 1.3.0, 1.4.0, 1.4.1, 1.4.2, 1.4.3a1, 1.4.3, 1.4.4, 1.4.5, 1.5.2, 1.5.3, 1.5.4, 1.5.5, 1.5.6, 1.5.7, 1.5.8)
ERROR: No matching distribution found for torch-cluster==latest+cu101

Looking in links: https://pytorch-geometric.com/whl/torch-1.7.1.html
ERROR: Could not find a version that satisfies the requirement torch-spline-conv==latest+cu101 (from versions: 0.1.0, 1.0.0, 1.0.1, 1.0.3, 1.0.4, 1.0.5, 1.0.6, 1.1.0, 1.1.1, 1.2.0, 1.2.1)
ERROR: No matching distribution found for torch-spline-conv==latest+cu101

I'm using python 3.7.1, cuda version is 1.0.1 and I'm using colab's GPUs...
Thanks in advance for your help !

Author

ameya98 commented on Feb 27, 2021

edited

I believe this is due to upstream changes in pip v20.3 and later. I had similar issues elsewhere.

Edit: Should be fixed now! I used the solution at pyg-team/pytorch_geometric#1876.
Let me know if it works!

nshl99 commented on Mar 2, 2021

edited

Hi, I've just tried this, but I'm still getting this same error:

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

It's strange as I managed to get it to work this morning without using this method, but that doesn't work too now.
Any ideas?

I ran it again but this time on the GPU, and I now get this error when trying to import torch_geometric:
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 11.0. Please reinstall the torch_sparse that matches your PyTorch install.

I installed torch again with:
!pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

Then proceeded to install all the torch_geometric modules using torch1.7.1+cu110, and this seems to work fine

Author

ameya98 commented on Mar 3, 2021

Interesting, I'll take a look.

I believe this is due to upstream changes in pip v20.3 and later. I had similar issues elsewhere.

Edit: Should be fixed now! I used the solution at rusty1s/pytorch_geometric#1876.
Let me know if it works!

Works thank you very much !

Huge thanks for this works for me without any changes needed.

!pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

This works for me as I was having problem to solve the problem installing torch-scatter

Author

ameya98 commented on Jun 3, 2021

Not sure I understand, is something not working? @ali1810

Thank you! Works perfectly!

homerjed commented on Jul 7, 2021

edited

I found only this worked, and it worked in under 5x the time:

!pip install -q torch-scatter -f https://pytorch-geometric.com/whl/torch-1.9.0+cu102.html
!pip install -q torch-sparse -f https://pytorch-geometric.com/whl/torch-1.9.0+cu102.html
!pip install -q torch-cluster -f https://pytorch-geometric.com/whl/torch-1.9.0+cu102.html
!pip install -q torch-geometric

I found only this worked, and it worked in under 5x the time:

!pip install -q torch-scatter -f https://pytorch-geometric.com/whl/torch-1.9.0+cu102.html
!pip install -q torch-sparse -f https://pytorch-geometric.com/whl/torch-1.9.0+cu102.html
!pip install -q torch-cluster -f https://pytorch-geometric.com/whl/torch-1.9.0+cu102.html
!pip install -q torch-geometric

@homerjed Works well+1

Author

ameya98 commented on Jul 11, 2021

Thanks @homerjed! Let me try to see if there's a way that will automatically find the latest package versions.

It worked for me. Thank you

This was massively helpful. Thanks!

Jiayi-Pan commented on Nov 3, 2021

edited

Colab has updated its cuda version recently, and ALL the solutions above seem DO NOT WORK.
TRY THIS ONE

# Add this in a Google Colab cell to install the correct version of Pytorch Geometric.
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://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-sparse -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-cluster -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-spline-conv -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-geometric

That also did not work with me,
I received this error OSError: libcudart.so.10.2: cannot open shared object file: No such file or directory

ved164 commented on Feb 1

Works for me. Thank you.

@Jiayi-Pan's code still works today. Thanks!

Code works fine. Thank you!

paulagd commented 12 days ago

This was very helpful! Still works today!

Colab has updated its cuda version recently, and ALL the solutions above seem DO NOT WORK. TRY THIS ONE

# Add this in a Google Colab cell to install the correct version of Pytorch Geometric.
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://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-sparse -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-cluster -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-spline-conv -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-geometric

This is not working currently on google colab. Can anyone help me out? Thanks in advance.

Jiayi-Pan commented yesterday

edited

Add this in a Google Colab cell to install the correct version of Pytorch Geometric.

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://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-sparse -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-cluster -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-spline-conv -f https://data.pyg.org/whl/torch-{TORCH}+{CUDA}.html
!pip install torch-geometric

Hi, it still works fine on my side at Colab as 2022/5/14.


About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK