使用Google Colab-使用GPU训练

一、基础操作

新建笔记本：

选用GPU：

显卡配置信息：

输入代码：

!nvidia-smi

 确实比CPU快点：

二、让自己联邦元实验放入运行

（1）将自己的代码文件传到“云端硬盘”

 （2）新建笔记本，修改设置为cuda：

 然后就可以看到硬盘中上传的自己的文件了：

复制文件路径，进入到此文件夹下。然后使用以下代码（注意开头感叹号必须有），在gpu上进行训练：

!python run.py --algo=fedmeta --eval_on_test_every=1 --dataset=femnist_p_0.2 --lr=0.001 --num_epochs=1 --model=cnn --clients_per_round=4 --batch_size=2 --data_format=pkl --num_rounds=100 --meta_algo=maml --outer_lr=0.0001 --result_prefix=./fedmeta_result --device=cuda --save_every=10

顺利运行，而且很快。

 但是，代码绝对有问题。虽然进行了一百论训练，但是，到20轮左右的时候就收敛了：