Skip to content

Commit 0f23d07

Browse files
mvsuspEliza Zhang
authored andcommitted
GPU fix (aws#117)
* Force framework libraries to re-install
1 parent ca0ee1b commit 0f23d07

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

docker/1.11.0/Dockerfile.cpu

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ WORKDIR /
6363
COPY $framework_installable .
6464
COPY $framework_support_installable .
6565

66+
<<<<<<< HEAD
6667
<<<<<<< HEAD
6768
RUN pip install --no-cache-dir -U \
6869
keras==2.2.4 \
@@ -73,6 +74,9 @@ RUN pip install --no-cache-dir -U \
7374
&& pip install --force-reinstall --no-cache-dir -U $framework_installable \
7475
=======
7576
RUN pip install --no-cache-dir $framework_installable \
77+
=======
78+
RUN pip install --no-cache-dir -U $framework_installable \
79+
>>>>>>> GPU fix (#117)
7680
$framework_support_installable\
7781
"sagemaker-tensorflow>=1.11,<1.12" \
7882
>>>>>>> Add distributed training support (#98)

docker/1.11.0/Dockerfile.gpu

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@ WORKDIR /
8484
COPY $framework_installable .
8585
COPY $framework_support_installable .
8686

87+
<<<<<<< HEAD
8788
<<<<<<< HEAD
8889
RUN pip install --no-cache-dir -U \
8990
keras==2.2.4 \
@@ -94,6 +95,9 @@ RUN pip install --no-cache-dir -U \
9495
&& pip install --force-reinstall --no-cache-dir -U $framework_installable \
9596
=======
9697
RUN pip install --no-cache-dir $framework_installable \
98+
=======
99+
RUN pip install --no-cache-dir -U $framework_installable \
100+
>>>>>>> GPU fix (#117)
97101
$framework_support_installable\
98102
"sagemaker-tensorflow>=1.11,<1.12" \
99103
>>>>>>> Add distributed training support (#98)

0 commit comments

Comments
 (0)