diff --git a/Dockerfile b/Dockerfile index cef944c7cc1ca49dd03a12499585a9ffda8fe6f1..3100e956d3f5975aaf4621cb740769f8f613fd02 100644 --- a/Dockerfile +++ b/Dockerfile @@ -54,7 +54,7 @@ RUN conda install -y python=${python_version} && \ pip install \ sklearn_pandas \ opencv-python \ - tensorflow-gpu==2.0.0 \ + tensorflow-gpu \ cntk-gpu && \ conda install \ bcolz \