Multiple GPUs with the OpenAi_ros framework

General Support
, ,
1

Hi,

I’m hoping to use the openai_ros framework for training Deep RL models and would like to know if we can use any distributed training methods with multiple GPUs. Currently, I’m only working with Pytorch with GPU:0 and would like to fully use the available GPU I have for the project.

Thank you

2

HI,

Of course, Tensorflow, PYTOrch and so on all of them have multiple GPU support:

https://pytorch.org/tutorials/beginner/former_torchies/parallelism_tutorial.html

1 Like
3

This topic was automatically closed 10 days after the last reply. New replies are no longer allowed.