torch-geometric本地whl安装

### 安装 `torch-geometric` 库 为了成功安装 `torch-geometric` 库并确保其正常工作，建议按照特定顺序来安装所需的依赖包。以下是详细的安装指南： #### 1. 确认 PyTorch 版本与 CUDA 配置 在开始之前，确认已正确安装所需版本的 PyTorch 并配置好相应的 CUDA 版本。例如，如果使用的是 Python 3.x 和 CUDA 11.1，则应先通过如下命令安装合适的 PyTorch 版本： ```bash pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu111 ``` #### 2. 安装必要的扩展包 接着，依次安装四个主要的支持库 (`torch-scatter`, `torch-sparse`, `torch-cluster`, `torch-spline-conv`)，这些库提供了图神经网络操作的基础功能。对于 PyTorch 1.8.0 + cu111 组合来说，具体命令如下所示[^1]: ```bash pip install torch-scatter==2.0.8 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html pip install torch-sparse==0.6.9 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html pip install torch-cluster==1.5.9 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html pip install torch-spline-conv==1.2.1 -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html ``` #### 3. 安装 `torch-geometric` 当所有前置条件满足后，可以通过下面的方式安装目标版本的 `torch-geometric`[^3]： ```bash pip install torch-geometric==2.0.1 -f https://data.pyg.org/whl/torch-1.9.0+cu111.html ``` 注意，在某些情况下，如果不指定 `-f` 参数后面的 URL 或者不精确匹配 PyTorch/CUDA 版本组合，可能会遇到兼容性问题。 #### 4. 验证安装 完成以上步骤后，可以在 Python 中导入模块以验证是否安装成功： ```python import torch_geometric print(torch_geometric.__version__) ```