【教程】安装DGL/PyG图神经网络编程环境

小锋学长生活大爆炸

已于 2025-02-25 10:50:51 修改

阅读量981

点赞数 4

分类专栏：图神经网络文章标签： pytorch 深度学习人工智能服务器 GPU linux

于 2024-06-24 19:29:21 首次发布

本文链接：https://blog.csdn.net/sxf1061700625/article/details/139935989

版权

图神经网络专栏收录该内容

15 篇文章

订阅专栏

转载请注明出处：小锋学长生活大爆炸[xfxuezhagn.cn]

如果本文帮助到了你，欢迎[点赞、收藏、关注]哦~

关于cuda的安装，可以看这个：

【教程】保姆级安装NVIDIA CUDA、CUDNN环境全纪录+解决SSH一段时间自动断开报Destination Host Unreachable_nvidia-driver-550 : 依赖: nvidia-dkms-550 (<= 550.54-CSDN博客文章浏览阅读1.2w次，点赞2次，收藏8次。非常全面且详细！_nvidia-driver-550 : 依赖: nvidia-dkms-550 (<= 550.54.14-1) 但是它将不会被安https://blog.csdn.net/sxf1061700625/article/details/127705588安装PyTorch：

pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121

pip install torch_geometric torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.3.0+cu121.html

安装DGL，其中根据你的cuda版本修改其中的数字(cu118)：

# If you have installed dgl-cuXX package, please uninstall it first.
pip install dgl==2.3.0 -f https://data.dgl.ai/wheels/torch-2.3/cu121/repo.html

pip install dglgo -f https://data.dgl.ai/wheels-test/repo.html

如果是PyG：

pip install torch_geometric

pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.3.0+cu121.html