Skip to content

Commit 6b1af9e

Browse files
committed
doc: Add info about GPU Docker images
1 parent f3c831e commit 6b1af9e

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

lib/AI/TensorFlow/Libtensorflow/Manual/Quickstart.pod

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,17 @@ C<http://127.0.0.1:8888/> in order to connect to the Jupyter Notebook interface
9494
via the web browser. In the browser, click on the C<notebook> folder to access
9595
the notebooks.
9696

97+
If using the GPU Docker image for NVIDIA support, make sure that the
98+
L<TensorFlow Docker requirements|https://www.tensorflow.org/install/docker#tensorflow_docker_requirements>
99+
are met and that the correct flags are passed to C<docker run>, for example
100+
101+
C<<
102+
docker run --rm --gpus all [...]
103+
>>
104+
105+
More information about NVIDIA Docker containers can be found in the user guide
106+
for the L<NVIDIA Container Toolkit|https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/user-guide.html>.
107+
97108
=begin :list
98109

99110
= C<latest>: base image with only C<libtensorflow> installed.
@@ -107,6 +118,8 @@ L<InferenceUsingTFHubEnformerGeneExprPredModel|AI::TensorFlow::Libtensorflow::Ma
107118
= C<latest-nb-omnibus>: image containing dependencies for both of the above
108119
notebooks.
109120

121+
= C<latest-gpu-nb-omnibus>: same as C<latest-nb-omnibus> but with NVIDIA GPU support.
122+
110123
=end :list
111124

112125
=head1 SEE ALSO

0 commit comments

Comments
 (0)