PyTorch|【PyTorch】torch-geometric 安装 pytorch|深度学习|神经网络|自动

文章目录

1. 查看cuda版本
2. 安装依赖
3. 完成安装
4. 遇到的问题

任务简介：
由于 torch-geometric 的安装需要一定的依赖，所以直接 pip install torch_geometric 安装会出现问题，并且 torch-geometric 的版本需要和 cuda 版本对应。
本文记录 torch-geometric 的正确安装过程。
1. 查看cuda版本查看安装的pytorch对应的cuda版本，比如 cuda10.2 或 cpu 版本
代码：

import torch print(torch.version.cuda)

输出：

10.2

2. 安装依赖根据1.中的对应版本安装以下依赖

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html

其中，${TORCH} 和 ${CUDA} 分别替换为（1）中确认的版本。
${TORCH} 可选 {1.4.0, 1.5.0, 1.6.0, 1.7.0, 1.8.0}等，PyTorch 1.8.1 选 1.8.0；
${CUDA} 可选 {cpu，cu92，cu101，cu102，cu110}；
对应 torch 可设置版本可通过以下链接查看最新版本：
https://pytorch-geometric.com/whl/index.html
假设使用 PyTorch 1.8.1 cu102 版本，安装以上依赖为：

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

3. 完成安装上述依赖安装完成后，torch-geometric 即可正常安装，命令：

pip install torch-geometric

4. 遇到的问题我使用的是 torch1.8.1，在上述安装完成后，torch-geometric会默认安装 2.0.4版本，2.0.4版本在代码中会出现一些问题，重新安装 torch-geometric 1.7.2 后可正常使用。
【PyTorch|【PyTorch】torch-geometric 安装】以下是 torch1.8.1 对应能正常使用的依赖版本: