From aafd894efd96c2a7f72b5736fef6208cd811566a Mon Sep 17 00:00:00 2001
From: lnstadrum <21985366+lnstadrum@users.noreply.github.com>
Date: Wed, 27 Mar 2024 16:56:52 +0100
Subject: [PATCH] Updating TensorFlow dockerfile

---
 tensorflow.Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow.Dockerfile b/tensorflow.Dockerfile
index 3c7d093..b54f868 100644
--- a/tensorflow.Dockerfile
+++ b/tensorflow.Dockerfile
@@ -1,10 +1,10 @@
-FROM tensorflow/tensorflow:2.7.1-gpu
+FROM tensorflow/tensorflow:latest-gpu
 
 # add source code
 ADD . /opt/fastaugment
 
 # install cmake
-RUN apt install -y cmake
+RUN apt update && apt install -y cmake
 
 # compile
 RUN cd /opt/fastaugment/tensorflow &&\