Skip to content

Commit 44bf534

Browse files
authored
Add note about container toolkit
1 parent 9df724b commit 44bf534

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,8 @@ be rebuilt, all of the subsequent steps will also be rebuilt.
120120
version to run
121121
122122
If your model requires a GPU, be sure to expose it by adding `--runtime nvidia`
123+
or `--gpus all`. Note that your local machine will also need the
124+
[NVIDIA Container Toolkit](https://github.com/NVIDIA/nvidia-docker).
123125
124126
### Prepare and submit your model
125127

0 commit comments

Comments
 (0)