代码链接:https://github.com/harvardnlp/botnet-detection
环境配置:
python 3.7
torch 1.2
torchvision==0.4.0
torch-scatter == 1.3.1
torch-cluster == 1.4.4
torch-sparse == 0.4.3
torch-spline-conv == 1.1.1
torch-geometric == 1.3.2
版本间关系参考:https://blog.csdn.net/u014714362/article/details/108614868
安装路径:
torch && torchvision 下载:https://download.pytorch.org/whl/torch_stable.html
torch-scatter 下载:https://github.com/rusty1s/pytorch_scatter/releases?page=2
torch-cluster 下载:https://github.com/rusty1s/pytorch_cluster/releases
torch-sparse 下载:https://github.com/rusty1s/pytorch_sparse/releases
torch-spline-conv 下载:https://github.com/rusty1s/pytorch_spline_conv/releases
torch-geometric下载:https://github.com/pyg-team/pytorch_geometric/releases
下载压缩包后 解压 运行python setup.py install即可安装
pytorch-geometric官网:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
pytorch-geometric作者GitHub:https://github.com/rusty1s
注:
pytorch 1.2 对应的cuda版本:CUDA 10.0.130
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html