I have tried to reproduce the code

[phdmatamoros]

I have
tensorflow 1.15.0
tensorflow-gpu 1.15.2
tensorflow-probability 0.7.0
cudatoolkit 10.0.130
cudnn 7.3.1

but i have this error in "lstm_model.produce_embeddings(config, model_vae, data, sess)"

UnknownError: 2 root error(s) found.
(0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
[[{{node encoder/conv2d/Conv2D}}]]
(1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
[[{{node encoder/conv2d/Conv2D}}]]
[[encoder/code_mean/BiasAdd/_7]]
0 successful operations.
0 derived errors ignored.