# Can't build keras model after Tensorflow update from 2.2 to 2.3

**Posted by ebgoldst on Aug 20, 2020**

See the new workaround. Also, try `pip3 uninstall tensorflow` to revert back to the version installed by Lambda Stack. If the error message doesn’t change, try adding the following after importing tensorflow:

```python
for device in tf.config.experimental.list_physical_devices('GPU'):
    tf.config.experimental.set_memory_growth(device, True)
```

---

## Background
I have been using keras to train some models for the past few months, and everything has been running smoothly. A few days ago, we updated the workstation including a Tensorflow update that brought us from 2.2 to 2.3. Ever since then, I can no longer train my models. It gets to the line where I add ...