Skip to content

Commit 0e1d847

Browse files
committed
updated Dockerfile.gpu to install cuda-toolkit
1 parent 7aa35df commit 0e1d847

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tensorflow_serving/tools/docker/Dockerfile.gpu

+2
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,8 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machi
6060
&& apt-get clean && \
6161
rm -rf /var/lib/apt/lists/*;
6262

63+
RUN apt-get update && apt-get install -y cuda-toolkit-11-8
64+
6365
# Install TF Serving GPU pkg
6466
COPY --from=build_image /usr/local/bin/tensorflow_model_server /usr/bin/tensorflow_model_server
6567

0 commit comments

Comments
 (0)