问题描述
我给伯乐框架添加了一些其他模型,比如DirectAU等,然后发现这个可以使用GPU但是只能在0卡上使用,不能根据显卡空余空间调节gpu_id的值。
通过分析,我们定位到recbole.config.configurator.py
这里就是在配置device的时候,只是把cuda加进去了,而没有设计到通过gpu_id来设计这个cuda调控。可以做如下修改。
else torch.device("cuda:"+self.final_config_dict["gpu_id"])
将上面那一行换成上面这行。
然后问题解决。