FATE icon indicating copy to clipboard operation
FATE copied to clipboard

Published 20 hours ago • verified publisher icon FederatedAI

fate环境里面如何使用gpu呢

Open zzy484 opened this issue 1 year ago • 5 comments

zzy484 avatar Aug 09 '23 03:08 zzy484

是使用Homo-NN吗？还是什么算法呢

talkingwallace avatar Aug 09 '23 08:08 talkingwallace

是使用Homo-NN吗？还是什么算法呢

是使用homo-nn

zzy484 avatar Aug 09 '23 12:08 zzy484

在trainer_param里设置cuda就可以了，https://github.com/FederatedAI/FATE/blob/master/python/federatedml/nn/homo/trainer/fedavg_trainer.py#L42

talkingwallace avatar Aug 10 '23 02:08 talkingwallace

在trainer_param里设置cuda就可以了，https://github.com/FederatedAI/FATE/blob/master/python/federatedml/nn/homo/trainer/fedavg_trainer.py#L42

image 我这样添加之后，显示错误：ValueError: Cuda is not available on this machine 但是机器上面有设备号为0的GPU image 这是什么原因呢

zzy484 avatar Aug 10 '23 03:08 zzy484

torch版本问题 看一下torch是否为cpu版本 如果是的话可以卸载重装

talkingwallace avatar Aug 11 '23 09:08 talkingwallace