Federated Graph Neural Network

环境配置

1. 创建 Conda 虚拟环境

conda create -n XX python=3.8
conda activate XX

2. 安装 PyTorch 和 CUDA 12.1

pip install torch==2.4.0 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu121

3. 安装 PyTorch Geometric 相关依赖

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv -f https://data.pyg.org/whl/torch-2.4.0+cu121.html
pip install torch-geometric==2.6.1

4. 安装其他依赖

运行

conda activate XX
python main.py --dataset_name bace --data_path  --input_model_file  --generated_data_dir