hzwer/ECCV2022-RIFE

It is not using GPU

Pekgame opened this issue · 4 comments

I have NVIDIA RTX 3070 GPU and when I ran this command:
python inference_video.py --exp=2 --scale=1 --video=test_video.mp4

It wasn't using any of my GPU but a CPU instead.

me to

我也是这样的,很奇怪

怎么看在使用cpu

https://pytorch.org/get-started/locally/
image
try to install CUDA with this command