代码链接: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

https://developer.nvidia.com/cuda-toolkit-archive

posted on 2021-11-07 15:32  曹婷婷  阅读(86)  评论(0编辑  收藏  举报