关于伯乐框架(RecBole)某些模型不能调控GPU解决方案

问题描述

我给伯乐框架添加了一些其他模型，比如DirectAU等，然后发现这个可以使用GPU但是只能在0卡上使用，不能根据显卡空余空间调节gpu_id的值。
通过分析，我们定位到recbole.config.configurator.py

这里就是在配置device的时候，只是把cuda加进去了，而没有设计到通过gpu_id来设计这个cuda调控。可以做如下修改。

else torch.device("cuda:"+self.final_config_dict["gpu_id"])

将上面那一行换成上面这行。
然后问题解决。