update dev container with tensorflow

bigsk1 · Jan 24, 2024 · 99d7e7c · 99d7e7c
1 parent 319804e
commit 99d7e7c
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/_posts/2024-01-22-Docker-Dev-Container.md b/_posts/2024-01-22-Docker-Dev-Container.md
@@ -333,6 +333,20 @@ Tensor after addition:
 PyTorch and CUDA are working correctly!
 ```
 
+## Tensorflow
+
+[https://www.tensorflow.org/install/pip](https://www.tensorflow.org/install/pip)
+
+If needed you can install tensorflow with GPU use 
+
+```bash
+pip install tensorflow
+```
+and then check was installed correctly
+
+```bash
+python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
+```
 
 
 ## Github