Skip to content

Commit 5913b17

Browse files
authored
GPU fix (#117)
* Force framework libraries to re-install
1 parent a897135 commit 5913b17

File tree

4 files changed

+9
-8
lines changed

4 files changed

+9
-8
lines changed

docker/1.10.0/Dockerfile.cpu

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,9 +63,10 @@ ENV KMP_AFFINITY=granularity=fine,compact,1,0 KMP_BLOCKTIME=1 KMP_SETTINGS=0
6363

6464
RUN framework_installable_local=$(basename $framework_installable) \
6565
&& framework_support_installable_local=$(basename $framework_support_installable) \
66-
&& pip install --no-cache --upgrade $framework_installable_local \
67-
&& pip install $framework_support_installable_local \
68-
&& pip install "sagemaker-tensorflow>=1.10,<1.11" \
66+
&& pip install --no-cache --upgrade \
67+
$framework_installable_local \
68+
$framework_support_installable_local \
69+
"sagemaker-tensorflow>=1.10,<1.11" \
6970
\
7071
&& rm $framework_installable_local \
7172
&& rm $framework_support_installable_local

docker/1.10.0/Dockerfile.gpu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,9 @@ COPY $framework_support_installable .
6262
RUN framework_installable_local=$(basename $framework_installable) && \
6363
framework_support_installable_local=$(basename $framework_support_installable) && \
6464
\
65-
pip install --no-cache --upgrade $framework_installable_local && \
66-
pip install $framework_support_installable_local && \
67-
pip install "sagemaker-tensorflow>=1.10,<1.11" &&\
65+
pip install --no-cache --upgrade $framework_installable_local \
66+
$framework_support_installable_local \
67+
"sagemaker-tensorflow>=1.10,<1.11" &&\
6868
\
6969
rm $framework_installable_local && \
7070
rm $framework_support_installable_local

docker/1.11.0/Dockerfile.cpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ WORKDIR /
5353
COPY $framework_installable .
5454
COPY $framework_support_installable .
5555

56-
RUN pip install --no-cache-dir $framework_installable \
56+
RUN pip install --no-cache-dir -U $framework_installable \
5757
$framework_support_installable\
5858
"sagemaker-tensorflow>=1.11,<1.12" \
5959
\

docker/1.11.0/Dockerfile.gpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ WORKDIR /
7474
COPY $framework_installable .
7575
COPY $framework_support_installable .
7676

77-
RUN pip install --no-cache-dir $framework_installable \
77+
RUN pip install --no-cache-dir -U $framework_installable \
7878
$framework_support_installable\
7979
"sagemaker-tensorflow>=1.11,<1.12" \
8080
\

0 commit comments

Comments
 (0)