Keras with tensorflow GPU


I haven't seen this information explicitly on the web rather than issues reported. When using a virtual environment where you have installed tensorflow-gpu and you install keras, keras will install the CPU tensorflow version. By doing this it won't be possible to use the GPU. Thus, the correct order to use keras with the GPU tensorflow version is:

  1. Install keras
  2. Install tensorflow-gpu


After this, the default backend for keras will be tensorflow and will work with the GPU.

To validate if you are using the GPU [1] :
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))


Comentarios

Publicar un comentario

Entradas populares de este blog

Deep neural networks visualization

Imagen
In this publication I present a "toy tool" to see how neurons on a Deep Neural Network (DNN) modify the links between them. These links are also called Weights. For those with technical knowledge, the "bias" unit is not shown (Because of lack of time, I will upgrade it later). In addition I haven't implemented yet the predict option, I would probably add it later. Technical Notes: This DNN uses standard back propagation without optimizations such as Adam optimizer or regularization parameters.   I am using Math.js  for the matrix operations. There are some errors I haven't figured out yet how to solve them, specially for Nan values that will appear when showing the visualization (Sometimes happens, no worry =) ) The last layer, the output layer is a sigmoid layer by default. Maybe later I will implement a linear function for regression analysis. And it has been implemented for binary classification by the moment. The weights are randomly init...
Leer más

Google Chart Sankey Diagram online for non software developers

Imagen
I saw lots of tutorials of how to create a Sankey diagram but I couldn't find a quick way to see my data without installing or configuring parameters. Thus, here I present a Sankey Diagram to visualize your data. In case you are a software developer this may result to be something simple for you, but in case you don't have programming skills but want a Sankey diagram, this post is for you. You have to upload a CSV file, comma separated in the format: NodeA, NodeB, Weight An example of a CSV file could be: Root A,Branch X,3 Root B,Branch X,8 Root B,Branch Y,2 Root B,Branch Z,3 Branch X,End 1,19 Branch X,End 2,2 Branch X,End 3,4 Branch Y,End 1,10 Branch Y,End 4,17 Root B,Branch ZX,4 Branch ZX,End 1,1 This will create the following Graph: Instructions of use: Way one Click on Chose File and search for your CSV file. In the text field you will see your data loaded. You should see "NodeX1,NodeY1,N1;NodeX2,NodeY2,N2... Click in the button Process ...
Leer más

Tensorflow Batch norm not working

Have you experienced that when using  tf.contrib.layers.batch_norm during test time, the performance is diminished or is completely a mess? If yes, this post is for you. Quick summary: 1) As suggested in [1]:   you have do add this additional code:   update_ops = tf . get_collection ( tf . GraphKeys . UPDATE_OPS )   with tf . control_dependencies ( update_ops ):     train_op = optimizer . minimize ( loss ) 2) You have to use a placeholder to indicate that you are training or not, so you put this in the batch_norm param  is_training = your_place_holder, True for training. 3) If after this it doesn't show a good behavior, change the default  decay = 0.999, to a lower value such as  0.5. Other way to apply batch norm is presented in [2], for a manual setup. Nonetheless, if by using this second approach you still have a bad performance, use the point 3). If you want to use this second approach with Convolutional Neural Netw...
Leer más