search

ZhengyiLuo / PHC

Official Implementation of the ICCV 2023 paper: Perpetual Humanoid Control for Real-time Simulated Avatars
https://zhengyiluo.github.io/PHC/
Other
480 stars 42 forks source link

Reduce Use of GPU Memory #58

Closed yw0208 closed 3 months ago

yw0208 commented 3 months ago

Hi, I added some complex scene meshes to the simulated environment. Then, the GPU memory is not enough. Is there any way to reduce the memory PHC needs? For example, could I delete some unnecessary steps? All I need is to keep the function of motion imitation. Looking forward to your reply. Thanks!

yw0208 commented 3 months ago

Does PHC support to be run on multiple GPUs? If it does, how should I set it?

ZhengyiLuo commented 3 months ago

To reduce GPU memory, the easiest thing to do is to reduce the number of envs. Try env.num_envs=1024.

I have never tried training PHC with multiple GPUs; it's certainly possible.