How to run these models purely on CPU. I have installed tensor-flow only for CPU based inference but when I analyze resources consumption through nvitop. It is shown that it consumes GPU memory for inference. How can I change this ?? Is there any tag like --cpu for CPU-based inference only. Or is there any other way to make GPU consumption as 0 in the code ?
How to run these models purely on CPU. I have installed tensor-flow only for CPU based inference but when I analyze resources consumption through nvitop. It is shown that it consumes GPU memory for inference. How can I change this ?? Is there any tag like --cpu for CPU-based inference only. Or is there any other way to make GPU consumption as 0 in the code ?