在 Rayune 上使用 GPU 或 CPU

Question

我的机器中有 1 个 GPU 和 32 个 CPU。 Ray 中可以单独使用它们吗？例如，一个任务分配有 1 个 CPU，另一任务分配有 1 个 GPU？ 如果我用

    tune.run(trainer_fn, 
             num_samples=32,
             resources_per_trial={"GPU": 1, "CPU": 1},
             ...

只运行一列火车，因为只有一个 GPU 可用。我知道我可以使用 0.1 GPU 之类的东西来运行多个列车，但我无法知道哪个应该使用 CPU，哪个应该使用 GPU。

Answer 1

最好的方法是使用 Ray Tune 实验 API，然后

tune.run_experiments

：https://docs.ray.io/en/latest/tune/api_docs/execution.html#tune-run-experiments

Answer 2

我正在尝试做同样的事情，但这个答案没有足够详细。如何让 Ray 使用 GPU 和任何可用的 CPU 来运行超参数搜索？我的机器是一台笔记本电脑，有 8 个 CPU 和 1 个 GPU。可以提供示例代码吗？