fast-style-transfer icon indicating copy to clipboard operation
fast-style-transfer copied to clipboard

Published 20 hours ago • verified publisher icon lengstrom

Question about the percentage of GPU usage

Open DanielMao2015 opened this issue 4 years ago • 5 comments

Hi, the code is successfully run on my GPU Nvidia 1060 Ti. However, it seems that the percentage of usage of GPU is low, which can be shown as the figure of GPU-Z. Is something going wrong with my training process? GPU-z

DanielMao2015 avatar Oct 26 '19 10:10 DanielMao2015

same in my 1050Ti

chenhan510 avatar Oct 28 '19 05:10 chenhan510

@chenhan510 Hi, did you solve the problem finally?

DanielMao2015 avatar Oct 29 '19 01:10 DanielMao2015

I am curious about this as well. Do we have a solution?

spot92 avatar Nov 12 '19 06:11 spot92

In the style.py file, I see a FRAC_GPU = 1 in line 22, but I don't think it is used anywhere at all. I am well below 1 (basically 0).

spot92 avatar Nov 12 '19 06:11 spot92

FRAC_GPU is the id of your GPU, which TensorFlow should determine automatically unless you force it to use a different GPU. Setting it to 0 will force TensorFlow to use CPU. Do note that TensorFlow uses CUDA primarily, not the primary chip, so you probably won't see exceptionally high usage on GPU-Z.

JEF1056 avatar Nov 12 '19 16:11 JEF1056