Cannot use GPU for training..

[lamhk]

Hi, i installed tensorflow-gpu and are able to use GPU for training on other reinforcement training projects (GPU has about 80% or above). However, I tried to specify the GPU but still the utilization of the GPU is still around 5-10%. Does anyone have experience? Thanks in advance.

import os os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"]="0" import keras

Regards, Lam

[rui-ren]

You can use this to update the gpu and Cuda:

pip install tf-nightly-gpu-2.0-preview==2.0.0.dev20191002