ceasona
ceasona
i download it from [this repository](https://github.com/jadewu/3D-Human-Face-Reconstruction-with-3DMM-face-model-from-RGB-image/tree/main/BFM)
默认是不支持GPU的,ray remote的时候需要添加gpu相关参数,其次,model和data都需要加载到gpu上才行
> 支持GPU和ray没有强依赖关系。隐语本身并不直接暴露ray给用户使用。 > > > 默认是不支持GPU的,ray remote的时候需要添加gpu相关参数,其次,model和data都需要加载到gpu上才行 不对吧,模型训练依赖于ray,ray控制着资源
> 有两种形态使用GPU。一种是直接使用ray调度GPU计算任务,由ray负责GPU的资源分配。另一种是仅把ray作为计算任务的拉起框架,计算任务是以进程方式运行,计算任务可以自行决定是否访问GPU。 目前SecretFlow的FL等算法其实是第二种,ray只是负责调度FL算法实体作为ray的actor启动,算法实体可以自行负责去访问GPU。 哦哦,感谢解答。我在backend用torch的情况下,不改ray配置的话好像没法使用到GPU