RuntimeError: CUDA out of memory

facebookresearch / pifuhd

High-Resolution 3D Human Digitization from A Single Image.

Other

9.49k stars 1.44k forks source link

RuntimeError: CUDA out of memory #157

Open HCW0727 opened 2 years ago

HCW0727 commented 2 years ago

Error message : RuntimeError: CUDA out of memory. Tried to allocate 36.00 MiB (GPU 0; 3.00 GiB total capacity; 2.42 GiB already allocated; 8.30 MiB free; 2.42 GiB reserved in total by PyTorch)

My Gpu memory is 3GB. Is there a way to run this program in my gpu?

remmel commented 2 years ago

README.md :

Note: At least 8GB GPU memory is recommended to run PIFuHD model.

Probably have to use your CPU : https://github.com/facebookresearch/pifuhd/issues/84#issuecomment-723693805