ubuntu20.04安装pyg

### 安装 PyG 库的详细说明 要在 Ubuntu 20.04 上安装 Python Geometric (PyG) 库，需要遵循一系列依赖项和特定版本的要求。以下是完整的指导： #### 1. 更新系统包管理器 在开始之前，建议更新系统的包管理器以确保所有基础工具是最新的： ```bash sudo apt update && sudo apt upgrade -y ``` #### 2. 安装必要的开发工具和库 某些编译过程可能需要额外的 C++ 编译器和其他支持库。可以通过以下命令安装这些依赖项： ```bash sudo apt install build-essential cmake git libboost-all-dev libsuitesparse-dev -y ``` #### 3. 创建并激活虚拟环境 为了避免与其他项目冲突，推荐使用 `virtualenv` 或 `venv` 来创建独立的 Python 环境。 ##### 使用 Virtualenv 创建虚拟环境 如果尚未安装 `virtualenv`，可以先通过以下命令安装它： ```bash pip install virtualenv ``` 接着创建一个新的虚拟环境（假设命名为 `pyg_env`），并激活该环境： ```bash virtualenv pyg_env source pyg_env/bin/activate ``` #### 4. 安装 PyTorch 和相关依赖 PyG 的运行依赖于 PyTorch。因此，在安装 PyG 之前，需先安装适合当前 CUDA 版本的 PyTorch。 对于 CPU-only 系统或者未启用 GPU 加速的情况，可以直接安装无 CUDA 支持的 PyTorch： ```bash pip install torch==2.0.0+cpu torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu ``` 如果有 NVIDIA 显卡，则需要根据显卡驱动程序匹配对应的 CUDA 版本。例如，CUDA 11.6 可以通过以下方式安装： ```bash pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116 ``` #### 5. 安装 PyG 及其扩展组件 完成 PyTorch 的安装后，继续安装 PyG 及其核心扩展模块。具体命令如下： ```bash pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.0.0+cu121.html pip install torch_geometric ``` 注意：URL 中的 `torch-2.0.0+cu121.html` 需要与已安装的 PyTorch 版本保持一致[^2]。 #### 6. 验证安装是否成功 进入 Python 解释器，尝试导入 PyG 并打印版本号来验证安装情况： ```python import torch_geometric print(torch_geometric.__version__) ``` 如果没有报错且能够正常显示版本号，则表明安装成功。 --- ### 注意事项 - 如果遇到兼容性问题，请确认所使用的 PyTorch 和 PyG 版本之间的对应关系。 - 对于较旧的操作系统或硬件设备，部分功能可能会受到限制。在这种情况下，可以选择降级至更稳定的组合版本[^3]。 ---