It is not using GPU
Pekgame opened this issue · 4 comments
Pekgame commented
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.
Dianli-Moli commented
me to
Dianli-Moli commented
我也是这样的,很奇怪
MikeAiJF commented
怎么看在使用cpu
tosunmon commented
https://pytorch.org/get-started/locally/
try to install CUDA with this command