【Mac版】基于pytorch环境安装PyG库+DGL库

最新推荐文章于 2024-08-02 17:00:53 发布

智慧的旋风

最新推荐文章于 2024-08-02 17:00:53 发布

阅读量2.7k

点赞数 4

分类专栏： python深度学习文章标签： anaconda python mac

本文链接：https://blog.csdn.net/weixin_41650348/article/details/109398819

版权

搭建GNN环境教程

1、创建python=3.6的conda虚拟环境

conda create -n env_gnn python=3.6

2、用conda安装pytorch1.6.0的cpu版本

pytorch官网安装命令。

conda install pytorch==1.6.0 torchvision==0.7.0 -c pytorch

import torch进行检验，没有问题。

3、pip安装cpu版本的PyG

pytorch-geometric(github项目)。
pytorch-geometric官方安装指南。

pip install torch-scatter==latest+cpu -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-sparse==latest+cpu -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-cluster==latest+cpu -f https://pytorch-geometric.com/whl/torch-