search

SvenGronauer / phoenix-drone-simulation

Simulation environments for the CrazyFlie quadrotor: Used for Reinforcement Learning and Sim-to-Real Transfer
MIT License
22 stars 9 forks source link

GPU usage #8

Open XilunZhangRobo opened 6 months ago

XilunZhangRobo commented 6 months ago

Hi thanks for the amazing codebase. I am wondering does the code use gpu during training? I did not find anything relevant on setting the device for training in the code.

mayhorizon commented 6 months ago

Hi, Yes I think it does. Otherwise you can set the training with multi cores and it doesn't run that slowly (20 minutes for 10,000 epochs with 32 cores)

SvenGronauer commented 5 months ago

Hi,

the training code examples are implemented for CPU usage. However, with some PyTorch code adjustments you can run the code also on a GPU.

Best, Sven