Блог

На моей машине доступны 1 графический процессор и 32 процессора. Возможно ли в Ray использовать их отдельно? Например, одной задаче выделяется 1 процессор, а другой задаче — 1 графический процессор? Если я использую

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

Это запускает только один поезд, потому что доступен только один графический процессор. Я знаю, что могу использовать что-то вроде 0.1 GPU, и это запускает несколько поездов, но у меня нет способа узнать, какой из них должен использовать CPU, а какой должен использовать GPU.

Лучший способ сделать это — использовать Ray Tune experiment API, а затем tune.run_experiments : https://docs.ray.io/en/latest/tune/api_docs/execution.html#tune-run-experiments