背景提要:需要借用GitHub的一个代码进行流预测,PyTorch RGCN (Link Prediction)
环境需求:安装如下版本的包
This code is lastly tested with:
python 3.7.x
pytorch 1.7.x
torch_geometric 1.7.x, with torch_scatter 2.0.6 and torch_sparse 0.6.9
1.安装指定版本的pytorch
官方链接:https://pytorch.org/get-started/previous-versions/
从官网查看所需命令:
针对CUDA 10.2版本,所需命令:pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2
pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.7.1+cu102.html
安装过程:
2. 安装指定版本的torch_geometric (方法无效)
官方链接:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
采用Installation via Pip Wheels方法安装
-
查看pytorch版本和cuda版本,分别是上述的1.7.1和10.2,
- 修改替换得到如下命令
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.7.1+cu102.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.7.1+cu102.html
pip install torch-geometric
以上方法失误
哈哈 后面发现Linux没有GPU,懒得搞CPU版本的了,换成Windows系统重新搞吧