0bserver07/One-Hundred-Layers-Tiramisu

Keras with tf-gpu installed but starts cpu training

codecolony opened this issue · 1 comments

Hi,

I have installed exactly as described in the instructions but the training always starts in CPU mode. I'm using tensorflow backend. Why do we need to install both tensorflow-gpu as well as tensorflow?

solved using...
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64