Pytorch Pytork geometric“；检测到Pytork和torch“U稀疏是用不同的CUDA版本编译的”；关于谷歌colab

我是pytorch geometric的新手，尝试将其安装到我的计算机上，但失败了，因此我尝试在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.

---

任何帮助都将不胜感激。

我已经更新了上一个问题的答案。希望现在能奏效。也不要忘记选择

GPU运行时

!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

---

我提出了以下代码片段，可以在Colab上安装PyTorch Geometric及其依赖项：

#将其添加到Google Colab单元中，以安装正确版本的Pyrotch Geometric。
进口火炬
def格式（版本）：
返回版本。拆分（“+”）[0]
火炬型=火炬型__
火炬=格式火炬版本（火炬版本）
def格式_cuda_版本（版本）：
返回'cu'+版本。替换（'.'，''）
CUDA_version=torch.version.CUDA
CUDA=格式\u CUDA\u版本（CUDA\u版本）
!pip安装火炬散布-fhttps://pytorch-geometric.com/whl/torch-{TORCH}+{CUDA}.html
!pip安装火炬稀疏-fhttps://pytorch-geometric.com/whl/torch-{TORCH}+{CUDA}.html
!pip安装火炬组-fhttps://pytorch-geometric.com/whl/torch-{TORCH}+{CUDA}.html
!pip安装火炬花键conv-fhttps://pytorch-geometric.com/whl/torch-{TORCH}+{CUDA}.html
!安装焊炬几何

查看它时会说“检查PyTorch是否安装了CUDA支持：

python-c”导入torch；打印（torch.cuda.is_available（））“

”。如果我键入

！python-c“import torch；print（torch.cuda.is_available（））”

在我的colab会话中，我得到了

False

，这是一个不好的迹象，该站点建议我应该得到

True

。但是，下面是：，如果我单击

菜单>运行时>更改运行时

，然后将硬件加速器更改为GPU，单击

保存

，运行与以前相同的代码（

！python-c“import torch；print（torch.cuda.is_available（））”

），我会得到一个

True

输出。如果您将硬件加速器更改为GPU，您是否会遇到以前遇到的错误？我自己尝试过，我认为如果启用GPU硬件加速器，它会起作用（我在安装您安装的所有东西之前启用了它，idk如果在之后启用它会发生什么情况）<代码>将torch_geometric.datasets导入为数据集在这种情况下运行，没有错误消息。