add back keras_applications & keras_preprocessing

st01cs · Sep 30, 2020 · c14b7fc · c14b7fc
1 parent bc3af14
commit c14b7fc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tensorflow-whl/Dockerfile b/tensorflow-whl/Dockerfile
@@ -72,7 +72,9 @@ RUN cp /usr/local/cuda-10.2/targets/x86_64-linux/include/cublas_api.h /usr/local
 RUN cd /usr/local/src && \
  git clone https://github.com/tensorflow/tensorflow && \
  cd tensorflow && \
- git checkout tags/v2.3.1
+ git checkout tags/v2.3.1 && \
+ pip install keras_applications --no-deps && \
+ pip install keras_preprocessing --no-deps
 
 # Create a TensorFlow wheel for CPU
 RUN cd /usr/local/src/tensorflow && \