Skip to content

Commit 89d4cd9

Browse files
authored
pip install tensorflow 1.8 in 1.8 cpu image (#57)
1 parent f8082e0 commit 89d4cd9

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

docker/1.8.0/final/py2/Dockerfile.cpu

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -7,23 +7,18 @@ ARG framework_support_installable=sagemaker_tensorflow_container-1.0.0.tar.gz
77

88
WORKDIR /root
99

10-
# Will install from pypi once packages are released there. For now, copy from local file system.
11-
COPY $framework_installable .
1210
COPY $framework_support_installable .
1311

14-
RUN framework_installable_local=$(basename $framework_installable) && \
15-
framework_support_installable_local=$(basename $framework_support_installable) && \
16-
\
17-
pip install --no-cache --upgrade $framework_installable_local && \
12+
RUN framework_support_installable_local=$(basename $framework_support_installable) && \
1813
pip install $framework_support_installable_local && \
19-
\
20-
rm $framework_installable_local && \
2114
rm $framework_support_installable_local
2215

16+
RUN pip install --no-cache --upgrade tensorflow==1.8.0
17+
2318
COPY sagemaker-tensorflow-extensions sagemaker-tensorflow-extensions
2419

2520
RUN cd sagemaker-tensorflow-extensions && \
26-
LD_LIBRARY_PATH=$(dirname $(find /usr/local/lib | grep "libmkl")):$LD_LIBRARY_PATH pip install . && \
21+
pip install . && \
2722
cd .. && \
2823
rm -rf sagemaker-tensorflow-extensions
2924

0 commit comments

Comments
 (0)