技巧分享|PyTorch Geometric（图卷积）环境安装 —— GPU 深度学习|人工智能|python|pyto

1. 安装PyTorch 进入 Pytorch 官网 pytorch.org，安装自己需要的版本。（我的是1.7.1版本cuda11.0）
注意：Pytotch 最低为 1.4.0 以上

文章图片

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

2. 安装PyTorch Geometric 【技巧分享|PyTorch Geometric（图卷积）环境安装 —— GPU】官方链接：Installation — pytorch_geometric 2.0.0 documentation

安装相关依赖库

pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cu110.html pip install --no-index torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.1+cu110.html pip install --no-index torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.1+cu110.html pip install --no-index torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.1+cu110.html

安装 torch-geometric

pip install torch-geometric -f https://data.pyg.org/whl/torch-1.7.1+cu110.html

注意：上述相关依赖库也可以进入官网下载后离线安装。网址：https://pytorch-geometric.com/whl/
3. 测试没有报错就说明安装成功了

python -c "import torch_geometric"

4. 附录 CPU 的安装，按以下方法安装(按照对应的版本号位置，修改成自己对应的版本号即可)。

pip install torch-scatter==2.0.4 -f https://pytorch-geometric.com/whl/torch-1.5.0.html pip install torch-sparse==0.6.3 -f https://pytorch-geometric.com/whl/torch-1.5.0.html pip install torch-cluster==1.5.4 -f https://pytorch-geometric.com/whl/torch-1.5.0.html pip install torch-spline-conv==1.2.0 -f https://pytorch-geometric.com/whl/torch-1.5.0.html pip install torch-geometric# 或则用下面方法安装最近更新的一个包 # pip install torch-scatter==latest+cpu -f https://pytorch-geometric.com/whl/torch-1.5.0.html

GPU 也可以按此方法，如下：

pip install torch-scatter==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.5.0.html pip install torch-sparse==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.5.0.html pip install torch-cluster==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.5.0.html pip install torch-spline-conv==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.5.0.html pip install torch-geometric

公众号：清纯世纪