File tree 1 file changed +4
-9
lines changed 1 file changed +4
-9
lines changed Original file line number Diff line number Diff line change @@ -7,23 +7,18 @@ ARG framework_support_installable=sagemaker_tensorflow_container-1.0.0.tar.gz
7
7
8
8
WORKDIR /root
9
9
10
- # Will install from pypi once packages are released there. For now, copy from local file system.
11
- COPY $framework_installable .
12
10
COPY $framework_support_installable .
13
11
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) && \
18
13
pip install $framework_support_installable_local && \
19
- \
20
- rm $framework_installable_local && \
21
14
rm $framework_support_installable_local
22
15
16
+ RUN pip install --no-cache --upgrade tensorflow==1.8.0
17
+
23
18
COPY sagemaker-tensorflow-extensions sagemaker-tensorflow-extensions
24
19
25
20
RUN cd sagemaker-tensorflow-extensions && \
26
- LD_LIBRARY_PATH=$(dirname $(find /usr/local/lib | grep "libmkl")):$LD_LIBRARY_PATH pip install . && \
21
+ pip install . && \
27
22
cd .. && \
28
23
rm -rf sagemaker-tensorflow-extensions
29
24
You can’t perform that action at this time.
0 commit comments