Google Colab: PyTorch Geometric Installation
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.
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?
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 !
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 •
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 •
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
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 •
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
@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.
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.
Recommend
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK