PyTorch Geometric安装
时间: 2025-02-25 19:43:19 浏览: 61
### 如何安装 PyTorch Geometric 库
#### 创建并激活 Conda 虚拟环境
为了确保依赖项不会冲突,建议创建一个新的 conda 环境来安装 PyTorch 和 PyTorch Geometric。
```bash
conda create -n pyg_env python=3.7
conda activate pyg_env
```
#### 安装 CUDA 工具包和其他基础库
根据目标硬件配置选择合适的 CUDA 版本。这里假设使用的是 CUDA 11.7:
```bash
conda install cudatoolkit=11.7
pip install scipy tensorboardx==2.5.1 tqdm normflows==1.4[^1]
```
#### 安装 PyTorch 及其几何扩展 (PyG)
对于特定版本的 PyTorch 和 PyTorch Geometric 的组合,可以从官方推荐的方式进行安装:
```bash
pip install torch-scatter==2.0.9 \
torch-sparse==0.6.15 \
torch-cluster==1.6.0 \
torch-spline-conv==1.2.2 \
torch-geometric==2.0.4 \
-f https://data.pyg.org/whl/torch-1.13.1+cu117.html[^2]
```
上述命令会从指定 URL 下载兼容于 PyTorch 1.13.1 和 CUDA 11.7 的预编译二进制文件,并安装必要的组件。
#### 验证安装成功
完成以上步骤之后,可以通过 Python 解释器验证 `torch` 和 `torch_geometric` 是否可以正常导入:
```python
import torch
print(torch.__version__)
import torch_geometric
print(torch_geometric.__version__)
```
如果没有任何错误提示,则说明安装过程顺利完成。
阅读全文
相关推荐


















