【教程】安装DGL/PyG图神经网络编程环境
原创
转载请注明出处：小锋学长生活大爆炸xfxuezhagn.cn 如果本文帮助到了你，欢迎点赞、收藏、关注哦~
关于cuda的安装，可以看这个：
https://cloud.tencent.com/developer/article/2162519
安装PyTorch，目前推荐2.2.1版本：
pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu118
pip install torch_geometric torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.1+cu118.html
安装DGL，其中根据你的cuda版本修改其中的数字(cu118)：
pip install dgl==2.1.0+cu118 -f https://data.dgl.ai/wheels/cu118/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.2.1+cu118.html
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
评论
登录后参与评论
推荐阅读