We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b23acfc commit 96a35a5Copy full SHA for 96a35a5
tensorflow_serving/tools/docker/Dockerfile.gpu
@@ -60,6 +60,8 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machi
60
&& apt-get clean && \
61
rm -rf /var/lib/apt/lists/*;
62
63
+RUN apt-get update && apt-get install -y cuda-toolkit-11-8
64
+
65
# Install TF Serving GPU pkg
66
COPY --from=build_image /usr/local/bin/tensorflow_model_server /usr/bin/tensorflow_model_server
67
0 commit comments