Updating TensorFlow dockerfile

lnstadrum · Mar 27, 2024 · aafd894 · aafd894
1 parent e82f65f
commit aafd894
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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 &&\