search

sczhou / ProPainter

[ICCV 2023] ProPainter: Improving Propagation and Transformer for Video Inpainting
https://shangchenzhou.com/projects/ProPainter/
Other
5.45k stars 645 forks source link

Not using Gpu ? #47

Open Avinash424a opened 11 months ago

Avinash424a commented 11 months ago

cpu usage at 80% and 0 for gpu , 1.2 gb vram usage while running the inference painter.py , takes about 20-25 minutes for a 1 second video at 720p

oranbox commented 11 months ago

the same problem. Maybe the right version of CUDA or 4090 is not working

thibaudart commented 10 months ago

I fixed it with:

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

H-Dempsey commented 10 months ago

I fixed the same issue by downgrading my version of python from 3.10 to 3.8.

teidenzero commented 6 months ago

I fixed it with:

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

This works for me, thank you! Can I ask you what was your thought process in order to come up with this solution?