pytorch 11.2 cuda11.3 Pytorch Geometric安装代码

安装 PyTorch 和 CUDA

在安装 PyTorch Geometric 之前，需要先安装 PyTorch 和 CUDA。可以使用以下命令安装 PyTorch：

# 安装 PyTorch
!pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu113/torch_stable.html

这里使用 `cu113` 是因为我当前使用的是 CUDA 11.3 版本，你可以根据自己的环境选择对应的版本。

安装 PyTorch Geometric

安装 PyTorch Geometric 需要使用 pip 命令，如下：

# 安装 PyTorch Geometric
!pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.9.0+cu113.html
!pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.9.0+cu113.html
!pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.9.0+cu113.html
!pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.9.0+cu113.html
!pip install torch-geometric

这里同样使用了 `cu113` 版本的 PyTorch，如果你使用的是其他版本的 PyTorch，也需要将上面链接中的版本号修改为对应的版本。

安装完成之后，可以使用以下代码验证是否安装成功：

import torch
import torch_geometric

print(torch.__version__)
print(torch_geometric.__version__)

如果输出了 PyTorch 和 PyTorch Geometric 的版本号，则说明安装成功。