diff --git a/docker/examples/Dockerfile b/docker/examples/Dockerfile index afd4c8d73..959c5a930 100644 --- a/docker/examples/Dockerfile +++ b/docker/examples/Dockerfile @@ -12,7 +12,8 @@ RUN pip install gym[atari] opencv-python-headless tensorflow lz4 keras pytest-ti RUN pip install -U h5py # Mutes FutureWarnings RUN pip install --upgrade bayesian-optimization RUN pip install --upgrade git+git://github.com/hyperopt/hyperopt.git -RUN pip install --upgrade sigopt nevergrad scikit-optimize hpbandster ConfigSpace lightgbm xgboost torch torchvision +RUN pip install ConfigSpace==0.4.10 +RUN pip install --upgrade sigopt nevergrad scikit-optimize hpbandster lightgbm xgboost torch torchvision RUN pip install -U tabulate mlflow RUN pip install -U pytest-remotedata>=0.3.1 RUN conda install pytorch-cpu torchvision-cpu -c pytorch diff --git a/docker/tune_test/Dockerfile b/docker/tune_test/Dockerfile index ae8355856..2b6aaef15 100644 --- a/docker/tune_test/Dockerfile +++ b/docker/tune_test/Dockerfile @@ -14,7 +14,8 @@ RUN conda remove -y --force wrapt RUN pip install gym[atari]==0.10.11 opencv-python-headless tensorflow lz4 keras pytest-timeout smart_open torch torchvision RUN pip install --upgrade bayesian-optimization RUN pip install --upgrade git+git://github.com/hyperopt/hyperopt.git -RUN pip install --upgrade sigopt nevergrad scikit-optimize hpbandster ConfigSpace lightgbm xgboost +RUN pip install ConfigSpace==0.4.10 +RUN pip install --upgrade sigopt nevergrad scikit-optimize hpbandster lightgbm xgboost RUN pip install -U mlflow RUN pip install -U pytest-remotedata>=0.3.1