司瓦图网站,编写app,逆冬seo,网站建设汇编材料目录 引言一、通过 Anaconda 安装二、通过 PyPi 安装三、从 Wheels 安装四、从 ROCm 安装五、从源代码安装5.1 确保 CUDA 环境设置正确5.1.1 检查 PyTorch 是否支持 CUDA5.1.2 设置 CUDA 环境变量5.1.3 验证 nvcc 是否可用 5.2 安装 PyTorch Geometric 所需软件包5.3 强制重新安… 目录 引言一、通过 Anaconda 安装二、通过 PyPi 安装三、从 Wheels 安装四、从 ROCm 安装五、从源代码安装5.1 确保 CUDA 环境设置正确5.1.1 检查 PyTorch 是否支持 CUDA5.1.2 设置 CUDA 环境变量5.1.3 验证 nvcc 是否可用 5.2 安装 PyTorch Geometric 所需软件包5.3 强制重新安装和调试 六、常见安装问题及解决方案6.1 出现 “undefined symbol: make_function_schema” 错误6.2 安装时出现 “ModuleNotFoundError” 错误6.3 CUDA 版本不匹配6.4 如何确认安装成功6.5 如何调整 CUDA 版本 七、安装流程图示总结 引言
PyTorch GeometricPyG是一个强大的图神经网络GNN库基于 PyTorch 进行构建专门用于处理图数据的深度学习任务。随着图神经网络在图像处理、社交网络分析、推荐系统等领域的广泛应用PyG 成为许多研究人员和开发者的首选工具之一。通过 PyG你可以轻松地实现图卷积、图采样、图分类等复杂操作从而构建出更为高效和准确的图神经网络模型。
然而安装 PyTorch Geometric 可能会因为系统配置、依赖库版本或硬件环境的不同而变得复杂。为了帮助你顺利安装 PyG并确保你的环境准备工作无误本博客将为你提供多种安装方法并详细介绍如何通过 Anaconda、PyPi、源代码等方式进行安装。同时我们还将讨论如何解决安装过程中常见的错误确保你能够快速、顺利地开始你的图神经网络之旅。
支持的 Python 版本
Python 3.9 到 3.12 是 PyTorch Geometric 当前支持的版本。如果你正在使用 Python 3.8 或更低版本建议升级 Python 版本以确保兼容性。
一、通过 Anaconda 安装
PyG 现在支持通过 Anaconda 在所有主流操作系统、PyTorch 和 CUDA 的组合上进行安装。如果你还没有安装 PyTorch请参考 PyTorch 官方文档 通过 conda 安装 PyTorch。确保你安装的 PyTorch 版本至少是 1.11.0。安装完成后可以通过以下命令安装 PyG
安装步骤 安装 PyTorch 首先确保你已经安装了 PyTorch。你可以使用以下命令通过 Anaconda 安装 PyTorch conda install pytorch -c pytorch如果你希望在 CUDA 上进行加速确保选择相应的 CUDA 版本。例如安装支持 CUDA 11.8 的 PyTorch conda install pytorch cudatoolkit11.8 -c pytorch安装 PyTorch Geometric 安装完 PyTorch 后可以通过以下命令来安装 PyTorch Geometric conda install pyg -c pyg注意目前Conda 包在 Windows 和 M1/M2/M3 Mac 上不可用。如果你是 Windows 或 Mac 用户建议使用 pip 安装。 指定 CUDA 版本 如果 Conda 安装时未正确识别你系统的 CUDA 版本你可以通过显式指定 CUDA 版本来进行安装。命令如下 conda install pyg**cu* -c pyg这个命令的作用是安装 pyg例如 PyTorch Geometric并指定相关的 CUDA 版本。我们可以进一步解析命令中的参数。 pyg**cu*此部分用于指定需要安装的库及其版本。 pyg 是指需要安装的包名称这里假设是 PyTorch Geometric根据你的环境包名称可能会有所不同。* 表示匹配所有版本。*cu* 用于指定 CUDA 版本其中 *cu* 表示匹配包含 CUDA 支持的包版本CUDA 版本会自动跟随选择。 -c pyg此参数指定了安装包的来源频道这里是 pyg 频道。你可以通过替换为其他合法频道来选择包源。
指定 CUDA 版本的实际示例
假设你需要安装 PyTorch Geometricpyg并且你希望使用特定版本的 PyTorch例如 1.13.0以及特定的 CUDA 版本例如 CUDA 11.7。你可以通过以下命令显式指定所需的 CUDA 版本
conda install pyg2.2.0pygcu117 -c pyg -c pytorch命令解析
pyg2.2.0指定安装 PyTorch Geometric 版本 2.2.0。pygcu117指定与 CUDA 11.7 兼容的版本。在 pyg 包中cu117 表示与 CUDA 11.7 兼容的构建版本。注意PyTorch 和 PyTorch Geometric 等包通常会有多个与不同 CUDA 版本兼容的版本。-c pyg指定使用 PyTorch Geometric 频道来获取包。-c pytorch指定使用 PyTorch 频道来获取 PyTorch 和其他相关依赖。
二、通过 PyPi 安装
如果你使用 pip 而非 conda你可以直接通过 PyPi 安装 PyTorch Geometric。
安装步骤 安装 PyTorch 在使用 PyG 之前确保已经安装了 PyTorch。可以通过以下命令进行安装 pip install torch安装 PyTorch Geometric 从 PyG 2.3 版本开始你可以仅通过 PyTorch 安装和使用 PyG无需额外的外部库。只需运行以下命令 pip install torch_geometric这种方式会自动安装必要的依赖并且适合大多数用户。
安装额外的库
PyTorch Geometric 提供了一些额外的库来扩展图神经网络的功能尤其是在大规模图和复杂模型训练时提供了各种加速和优化操作。以下是一些推荐的库
库名描述pyg-lib支持异构图神经网络Heterogeneous GNNs和图采样功能。torch-scatter高效的稀疏矩阵归约操作广泛用于稀疏数据的图处理。torch-sparse提供支持稀疏张量的操作提升计算效率。torch-cluster实现高效的图聚类操作适用于图的分区和聚类任务。torch-spline-conv支持基于样条的卷积SplineConv操作用于图中的边际学习。
这些库能够提升图神经网络训练的效率和灵活性。如果你希望在使用 PyG 时充分发挥其性能可以考虑安装这些额外的库
pip install pyg-lib torch-scatter torch-sparse torch-cluster torch-spline-conv这些库自带 CPU 和 GPU 内核实现基于 PyTorch 的 C/CUDA/hip(ROCm) 扩展接口。如果你刚开始使用 PyG可以从最小安装开始待实际需要时再安装额外的依赖库。
三、从 Wheels 安装
为了简化依赖库的安装过程PyTorch Geometric 提供了多种预构建的 wheel 包这些包支持多种操作系统、PyTorch 和 CUDA 版本的组合。你可以访问 PyG Wheel 下载页面 来选择与你的环境相匹配的版本。
警告目前 M1/M2/M3 的 Mac 不支持使用 Wheels 安装。请从源代码安装扩展包。
检查 PyTorch 版本和 CUDA 版本
在安装 PyG 的额外依赖之前需要确保你使用的 PyTorch 版本和 CUDA 版本正确请确保你的 PyTorch 版本至少是 1.13.0。你可以通过以下命令检查
python -c import torch; print(torch.__version__)2.5.0 # 例如显示 PyTorch 版本python -c import torch; print(torch.version.cuda)12.4 # 显示 CUDA 版本如果你已经安装了合适的 PyTorch 版本例如PyTorch 2.5.0 和 CUDA 12.4可以使用如下命令安装相关依赖库
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-${TORCH}${CUDA}.html其中 ${TORCH} 和 ${CUDA} 需要根据你的实际版本替换为你当前使用的 PyTorch 和 CUDA 版本。其中 ${TORCH} 和 ${CUDA} 应该被具体替换PyTorch和 CUDA 版本分别为
PyTorch2.4: ${TORCH}2.5.0 和 ${CUDA}cpu|cu118|cu121|cu124PyTorch2.4: ${TORCH}2.4.0 和 ${CUDA}cpu|cu118|cu121|cu124PyTorch2.3: ${TORCH}2.3.0 和 ${CUDA}cpu|cu118|cu121PyTorch2.2: ${TORCH}2.2.0 和 ${CUDA}cpu|cu118|cu121PyTorch2.1: ${TORCH}2.1.0 和 ${CUDA}cpu|cu118|cu121PyTorch2.0: ${TORCH}2.0.0 和 ${CUDA}cpu|cu117|cu118PyTorch1.13: ${TORCH}1.13.0 和 ${CUDA}cpu|cu116|cu117
例如 对于 PyTorch 2.5 和 CUDA 12.4 pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.5.0cu124.html对于 PyTorch 2.4 和 CUDA 11.8 pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.4.0cu118.html以下是一些常见的版本组合
PyTorch 版本CUDA 版本下载链接2.5.0cu124https://data.pyg.org/whl/torch-2.5.0cu124.html2.4.0cu118https://data.pyg.org/whl/torch-2.4.0cu118.html2.3.0cu121https://data.pyg.org/whl/torch-2.3.0cu121.html2.2.0cu118https://data.pyg.org/whl/torch-2.2.0cu118.html2.0.0cu117https://data.pyg.org/whl/torch-2.0.0cu117.html1.13.0cu116https://data.pyg.org/whl/torch-1.13.0cu116.html
注意还提供了旧版本的二进制文件PyTorch1.4.0、1.5.0、1.6.0、1.7.0/1.7.1、1.8.0/1.8.1、1.9.0、1.10.0/1.10.1/1.10.2、1.11.0、1.12.0/1.12.1、1.13.0/1.13.1、2.0.0/2.0.1、2.1.0/2.1.1/2.1.2、2.2.0/2.2.1/2.2.2 和 2.3.0/2.3.1按照相同的程序。对于较旧的版本您需要明确指定最新支持的版本号或通过安装以防止从源手动安装。您可以在此处查找最新支持的版本号。
pip install --no-index四、从 ROCm 安装
如果你使用的是 ROCm 平台可以参考 pyg-rocm-build 存储库这里提供了 PyTorch Geometric 相关的轮子包Whl 文件和安装详细说明。你可以根据该文档中的步骤进行安装。
如果你在安装过程中遇到问题欢迎在 GitHub Issues 提出问题我们会尽快解决。
五、从源代码安装
如果你发现我们的轮子包无法满足你的需求例如某些版本不支持你可以选择从源代码进行安装。以下是从源代码安装的详细步骤。
5.1 确保 CUDA 环境设置正确
如果你计划在 ROCm 上安装 PyTorch Geometric首先需要确保你的 CUDA 环境已正确配置尽管 ROCm 主要用于支持 AMD GPU但在一些兼容模式下CUDA 仍然有用。
5.1.1 检查 PyTorch 是否支持 CUDA
首先你需要确认你的 PyTorch 是否已经安装并启用了 CUDA 支持
python -c import torch; print(torch.cuda.is_available())True输出结果应为 True如果为 False说明 CUDA 支持未正确安装需要进行调试。
5.1.2 设置 CUDA 环境变量
在安装前确保 CUDA 环境正确设置。需要将 CUDA 的路径添加到系统的环境变量中确保安装过程能正确找到相关的库和工具。
假设 CUDA 安装路径为 /usr/local/cuda则可以通过以下命令设置环境变量
export PATH/usr/local/cuda/bin:$PATH
echo $PATH
# 输出应包含 /usr/local/cuda/bin/usr/local/cuda/bin:...接着设置 CPATH 环境变量
export CPATH/usr/local/cuda/include:$CPATH
echo $CPATH
# 输出应包含 /usr/local/cuda/include/usr/local/cuda/include:...如果你使用的是 Linux 系统还需要将 CUDA 的共享库路径添加到 LD_LIBRARY_PATH 中
export LD_LIBRARY_PATH/usr/local/cuda/lib64:$LD_LIBRARY_PATH
echo $LD_LIBRARY_PATH
# 输出应包含 /usr/local/cuda/lib64/usr/local/cuda/lib64:...
对于 macOS 用户则应将路径添加到 DYLD_LIBRARY_PATH
export DYLD_LIBRARY_PATH/usr/local/cuda/lib:$DYLD_LIBRARY_PATH
echo $DYLD_LIBRARY_PATH
# 输出应包含 /usr/local/cuda/lib/usr/local/cuda/lib:...5.1.3 验证 nvcc 是否可用
nvcc 是 NVIDIA CUDA 编译器你可以通过以下命令检查是否能从终端访问 nvcc
nvcc --version
# 输出应该显示当前安装的 CUDA 版本例如
# nvcc: NVIDIA (R) Cuda compiler driver
# Cuda compilation tools, release 11.8, V11.8.8911.8确保PyTorch并且系统 CUDA 版本匹配
python -c import torch; print(torch.version.cuda)11.85.2 安装 PyTorch Geometric 所需软件包
安装 PyTorch Geometric 及其依赖的扩展包。以下命令会安装 PyTorch Geometric 所需的各个子包包括 torch-scatter, torch-sparse, torch-cluster 等。
pip install --verbose githttps://github.com/pyg-team/pyg-lib.git
pip install --verbose torch_scatter
pip install --verbose torch_sparse
pip install --verbose torch_cluster
pip install --verbose torch_spline_conv这些命令会下载并编译各个依赖库确保你能够顺利使用 PyTorch Geometric。
5.3 强制重新安装和调试
在极少数情况下CUDA 或 Python 路径问题可能会导致安装失败。pip甚至可能表示安装成功但执行会崩溃。我们在常见问题解答小节中收集了常见的安装错误。如果常见问题解答无法帮助您解决问题请创建问题。在此之前请按照官方安装指南验证您的 CUDA 是否已正确设置。 你可以尝试强制重新安装相关软件包以避免由于缓存问题导致的错误
pip --force-reinstall --no-cache-dir
pip --verbose通过设置 --verbose 参数你可以查看详细的安装日志以帮助诊断安装过程中的问题。
六、常见安装问题及解决方案
6.1 出现 “undefined symbol: make_function_schema” 错误
如果你遇到如下错误
undefined symbol: make_function_schema通常原因如下
安装的 PyTorch 版本与 ${TORCH} 包指定的版本不匹配。安装的 CUDA 版本与 PyTorch 所使用的 CUDA 版本不一致。
解决方案
检查 PyTorch 版本
确认你安装的 PyTorch 版本及其支持的 CUDA 版本
python -c import torch; print(torch.__version__)输出示例
1.13.0接着检查 PyTorch 支持的 CUDA 版本
python -c import torch; print(torch.version.cuda)输出示例
11.8检查 CUDA 版本
使用以下命令检查系统中安装的 CUDA 版本
nvcc --version输出示例
nvcc: NVIDIA (R) Cuda compiler driver
Cuda compilation tools, release 11.8, V11.8.89确保 PyTorch 版本的 CUDA 与系统安装的 CUDA 版本一致。
重新安装并清除缓存
若发现版本不匹配强制重新安装 PyTorch 及相关依赖并清除缓存
pip install --force-reinstall --no-cache-dir torch
pip install --force-reinstall --no-cache-dir torch_scatter torch_sparse torch_cluster若仍然存在问题可以尝试从源代码重新安装相关扩展包
pip install --force-reinstall --no-cache-dir
pip install --verbose6.2 安装时出现 “ModuleNotFoundError” 错误
遇到 ModuleNotFoundError 错误时通常是由于 PyG 的依赖库没有正确安装。可以尝试重新安装必要的依赖库
pip install torch-scatter torch-sparse torch-cluster torch-spline-conv6.3 CUDA 版本不匹配
如果出现以下错误
CUDA version mismatch表示 PyTorch 或 PyTorch Geometric 版本与 CUDA 版本不兼容。No module named torch_scatter 或类似错误可能是安装了不兼容的版本。
解决方案
确保安装了与系统 CUDA 版本兼容的 cudatoolkit。使用的 PyTorch 和 PyTorch Geometric 版本与 CUDA 版本必须匹配。
6.4 如何确认安装成功
安装完 PyG 后运行以下代码验证安装是否成功
import torch
import torch_geometric
print(torch.__version__)
print(torch_geometric.__version__)如果没有报错且版本号正确则说明安装成功。
6.5 如何调整 CUDA 版本
如果你的系统 CUDA 版本与当前版本不匹配你可以选择不同版本的 CUDA 进行安装。例如若要使用 CUDA 11.6可以按如下方式安装
conda install pytorch1.13 torchvision torchaudio cudatoolkit11.6 -c pytorch
conda install pyg**cu116 -c pyg这样你可以根据自己的硬件和软件环境选择合适的 CUDA 版本进行安装。
七、安装流程图示
为了帮助更直观地理解整个安装流程以下是 PyTorch Geometric 安装的简化流程图 #mermaid-svg-qJjZDlTt78kHVDuq {font-family:"trebuchet ms",verdana,arial,sans-serif;font-size:16px;fill:#333;}#mermaid-svg-qJjZDlTt78kHVDuq .error-icon{fill:#552222;}#mermaid-svg-qJjZDlTt78kHVDuq .error-text{fill:#552222;stroke:#552222;}#mermaid-svg-qJjZDlTt78kHVDuq .edge-thickness-normal{stroke-width:2px;}#mermaid-svg-qJjZDlTt78kHVDuq .edge-thickness-thick{stroke-width:3.5px;}#mermaid-svg-qJjZDlTt78kHVDuq .edge-pattern-solid{stroke-dasharray:0;}#mermaid-svg-qJjZDlTt78kHVDuq .edge-pattern-dashed{stroke-dasharray:3;}#mermaid-svg-qJjZDlTt78kHVDuq .edge-pattern-dotted{stroke-dasharray:2;}#mermaid-svg-qJjZDlTt78kHVDuq .marker{fill:#333333;stroke:#333333;}#mermaid-svg-qJjZDlTt78kHVDuq .marker.cross{stroke:#333333;}#mermaid-svg-qJjZDlTt78kHVDuq svg{font-family:"trebuchet ms",verdana,arial,sans-serif;font-size:16px;}#mermaid-svg-qJjZDlTt78kHVDuq .label{font-family:"trebuchet ms",verdana,arial,sans-serif;color:#333;}#mermaid-svg-qJjZDlTt78kHVDuq .cluster-label text{fill:#333;}#mermaid-svg-qJjZDlTt78kHVDuq .cluster-label span{color:#333;}#mermaid-svg-qJjZDlTt78kHVDuq .label text,#mermaid-svg-qJjZDlTt78kHVDuq span{fill:#333;color:#333;}#mermaid-svg-qJjZDlTt78kHVDuq .node rect,#mermaid-svg-qJjZDlTt78kHVDuq .node circle,#mermaid-svg-qJjZDlTt78kHVDuq .node ellipse,#mermaid-svg-qJjZDlTt78kHVDuq .node polygon,#mermaid-svg-qJjZDlTt78kHVDuq .node path{fill:#ECECFF;stroke:#9370DB;stroke-width:1px;}#mermaid-svg-qJjZDlTt78kHVDuq .node .label{text-align:center;}#mermaid-svg-qJjZDlTt78kHVDuq .node.clickable{cursor:pointer;}#mermaid-svg-qJjZDlTt78kHVDuq .arrowheadPath{fill:#333333;}#mermaid-svg-qJjZDlTt78kHVDuq .edgePath .path{stroke:#333333;stroke-width:2.0px;}#mermaid-svg-qJjZDlTt78kHVDuq .flowchart-link{stroke:#333333;fill:none;}#mermaid-svg-qJjZDlTt78kHVDuq .edgeLabel{background-color:#e8e8e8;text-align:center;}#mermaid-svg-qJjZDlTt78kHVDuq .edgeLabel rect{opacity:0.5;background-color:#e8e8e8;fill:#e8e8e8;}#mermaid-svg-qJjZDlTt78kHVDuq .cluster rect{fill:#ffffde;stroke:#aaaa33;stroke-width:1px;}#mermaid-svg-qJjZDlTt78kHVDuq .cluster text{fill:#333;}#mermaid-svg-qJjZDlTt78kHVDuq .cluster span{color:#333;}#mermaid-svg-qJjZDlTt78kHVDuq div.mermaidTooltip{position:absolute;text-align:center;max-width:200px;padding:2px;font-family:"trebuchet ms",verdana,arial,sans-serif;font-size:12px;background:hsl(80, 100%, 96.2745098039%);border:1px solid #aaaa33;border-radius:2px;pointer-events:none;z-index:100;}#mermaid-svg-qJjZDlTt78kHVDuq :root{--mermaid-font-family:"trebuchet ms",verdana,arial,sans-serif;}#mermaid-svg-qJjZDlTt78kHVDuq .watermark*{fill:#fff!important;stroke:none!important;font-size:15px!important;opacity:0.8!important;}#mermaid-svg-qJjZDlTt78kHVDuq .watermark span{fill:#fff!important;stroke:none!important;font-size:15px!important;opacity:0.8!important;} 是 否 是 否 CSDN 2136 安装 Python 环境 安装 PyTorch 检查 CUDA 配置 安装 PyTorch Geometric 是否需要额外库? 安装 pyg-lib 等依赖 完成安装 测试安装是否成功 开始使用 PyTorch Geometric 可选: 安装其他扩展库 检查并解决常见问题 问题解决? 成功安装 处理安装错误 查看日志 根据日志修复问题 CSDN 2136 该流程图展示了从安装 PyTorch 到解决常见问题的整个过程帮助用户理解每个步骤的关系。 总结
安装 PyTorch GeometricPyG并不复杂但需要根据不同的环境和需求选择适合的安装方法。通过 Conda 或 PyPi 安装是最为常见且简便的方式适合大多数用户。而对于需要更多定制或特定版本的用户直接从源代码安装或通过 Wheels 安装也是不错的选择。无论你采用何种方式确保 CUDA 环境正确配置以及安装所需的依赖包都是顺利安装 PyG 的关键。
在安装过程中用户可能会遇到一些常见的错误例如 “undefined symbol: make_function_schema” 或 “ModuleNotFoundError” 错误。本文提供了详细的故障排查和解决方案帮助你快速识别并解决这些问题。最终随着正确安装和配置你可以开始使用 PyTorch Geometric进行高效的图神经网络建模和实验推动你的项目和研究向前发展。
通过本文的指导你应该能根据自己的环境需求选择最合适的安装方式并确保 PyTorch Geometric 在你的机器上成功运行开始图数据分析的探索旅程。
