CUDA 下 DGL 库安装

Posted 2022-07-14 叶庭云

在复现 Beta 小波图神经网络（BWGNN），图结构上的异常检测时，需要安装 DGL 库：

conda install -c dglteam dgl-cuda11.1

pip install sympy
pip install argparse
pip install scikit-learn -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
unzip autodl-tmp/Rethinking-Anomaly-Detection-master.zip

pip install psutil

结果如下：

psutil 是 Python 一个跨平台库能够轻松实现获取系统运行的进程和系统利用率（包括 CPU、内存、磁盘、网络等信息。它主要用来做系统监控，性能分析，进程管理。差这个 psutil 库，再 pip install 安装上即可，跑通了代码，测试结果如下：

References：

• pytorch系列（三）——cuda下DGL库安装
• DGL官网 DGL Install
• ICML 2022 | 基于结构化数据的异常检测再思考: 我们究竟需要怎样的图神经网络？