Catherine_zhilin

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

https://github.com/AutoML-Research/F2GNN

 git clone https://github.com/Shudeng/Point-GNN.pytorch --recursive


pip install https://github.com/rm-hull/luma.core/archive/master.zip
pip install https://github.com/rm-hull/luma.oled/archive/master.zip

pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

 pip install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.12.0+cu113.html


(F2SAGE) python train_search.py --data wisconsin --gpu 0 --agg sage --temp 0.001 --arch_learning_rate 0.01 --epochs 400 --learning_rate 0.02

(F2GAT) python train_search.py --data wisconsin --gpu 0 --agg gat --temp 0.001 --arch_learning_rate 0.01 --epochs 400 --learning_rate 0.02

(F2GNN) python train_search.py --data wisconsin --gpu 0 --search_agg True --temp 0.001 --arch_learning_rate 0.01 --epochs 400 --learning_rate 0.02

(Random SAGE) python train_search.py --data wisconsin --gpu 0 --agg sage --algo random --temp 0.001 --arch_learning_rate 0.01 --epochs 400 --learning_rate 0.02 --alpha_mode train --random_epoch 100

 

python fine_tune.py --data wisconsin --gpu 0  --hyper_epoch 30  --arch_filename  exp_res/wisconsin_sage.txt --cos_lr --layer_norm True 

 

 

 

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric




pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

posted on 2022-09-06 10:05  kkkshiki  阅读(449)  评论(0编辑  收藏  举报