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