search

ravisvi / super-resolution-videos

Applying SRGAN technique implemented in https://github.com/zsdonghao/SRGAN on videos to super resolve them.
The Unlicense
110 stars 33 forks source link

Can I use GPU? #5

Closed kokosandim closed 4 years ago

kokosandim commented 5 years ago

I have gtx 1070, how I can run this render on gpu? I dont find where I can specify this

ravisvi commented 5 years ago

Hi @kokosandim,

Hmm. The answer is mostly no.

You can use your GPU to train the models. This happens automatically if you have tensor flow installed with CUDA.

If you are talking about using the existing models in this repository and super-resolving videos using the supervideo.py then there is currently no way of specifying to the code base to use the GPU. However, if you are using tensor flow with CUDA and if your FFmpeg binary is compiled with the required support for NVIDIA libraries then, your computer should automatically use the graphics card I assume. But I'm not sure of this. Please have a look at this link here: https://developer.nvidia.com/ffmpeg

Make sure you are using the right FFmpeg binary after you get the one from above.