建议torch-geometric不要装最新版本,建议装1.2.1
-
首先切换到CMD,打开python交互环境
-
输入,查看你所安装的pytorch版本
import torch print(torch.__version__)
我的当前pytorch版本是>>> 1.6.0
-
安装依赖包及
torch-geometric
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 pip install torch-geometric
在那里${CUDA},并${TORCH}应通过特定CUDA版本(被替换cpu,cu92,cu101,cu102,cu110)和PyTorch版本(1.4.0,1.5.0,1.6.0,1.7.0)分别。例如,对于PyTorch 1.7.0和CUDA 11.0,键入:
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu110.html pip install torch-geometric
在安装
torch-cluster
报错,提示安装不上,解决办法如下:
1.打开下载网址为https://pytorch-geometric.com/whl/,选择相应的torch
版,本人的torch版本为1.6.0+cpu
2. 同时本人的python版本为3.7.7,则选择对应的torch-cluster
进行下载.whl
文件进行本地安装。
3. 将上一步下载的.whl
文件执行
pip install 下载的.whl文件路径+.whl文件
即可
4. 安装成功
-
检查安装结果
import torch import torch_geometric
安装成功,结束!