Update Dockerfile to v0.8.0

HazyResearch · Apr 11, 2020 · eb06245 · eb06245
1 parent c4f8d95
commit eb06245
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -22,11 +22,13 @@ ENV VIRTUAL_ENV=/home/user/.venv
 RUN python -m venv $VIRTUAL_ENV
 ENV PATH="$VIRTUAL_ENV/bin:$PATH"
 
-
-ARG FONDUER_VERSION=
-# Install python packages
 # Set --build-arg FONDUER_VERSION=X.X.X to install a specific version of
 # Fonduer, otherwise the lastest version is installed.
+ARG FONDUER_VERSION=
+
+# Install python packages
 RUN pip install \
-    https://download.pytorch.org/whl/cpu/torch-1.1.0-cp37-cp37m-linux_x86_64.whl \
+    torch==1.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html \
+    # https://github.com/HazyResearch/fonduer/issues/390
+    "tensorboard<2.0.0,>=1.14.0" "scikit-learn<0.22.0,>=0.20.2" \
     fonduer${FONDUER_VERSION:+==${FONDUER_VERSION}}