Skip to content

gpu limit · Changes

Page history
Update gpu limit authored by Daniele Jahier Pagliari's avatar Daniele Jahier Pagliari
Hide whitespace changes
Inline Side-by-side
gpu-limit.md
View page @ 47e4e1d2
...@@ -13,6 +13,7 @@ If you use TensorFlow 2.0 (or Keras with a TensorFlow backend), you can limit th ...@@ -13,6 +13,7 @@ If you use TensorFlow 2.0 (or Keras with a TensorFlow backend), you can limit th
```python ```python
import tensorflow as tf import tensorflow as tf
megabytes = 2048
gpus = tf.config.experimental.list_physical_devices('GPU') gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus: if gpus:
for gpu in gpus: for gpu in gpus:
......