Skip to content

Commit c14b7fc

Browse files
committed
add back keras_applications & keras_preprocessing
1 parent bc3af14 commit c14b7fc

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tensorflow-whl/Dockerfile

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,9 @@ RUN cp /usr/local/cuda-10.2/targets/x86_64-linux/include/cublas_api.h /usr/local
7272
RUN cd /usr/local/src && \
7373
git clone https://github.com/tensorflow/tensorflow && \
7474
cd tensorflow && \
75-
git checkout tags/v2.3.1
75+
git checkout tags/v2.3.1 && \
76+
pip install keras_applications --no-deps && \
77+
pip install keras_preprocessing --no-deps
7678

7779
# Create a TensorFlow wheel for CPU
7880
RUN cd /usr/local/src/tensorflow && \

0 commit comments

Comments
 (0)