We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7aa35df commit 0e1d847Copy full SHA for 0e1d847
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