Update README.md

allenai · May 22, 2020 · d21af08 · d21af08
1 parent 8ce4d0f
commit d21af08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -98,4 +98,4 @@ Now run the following command to get an environment that will run on either the
    docker run --rm -v $HOME/.allennlp:/root/.allennlp allennlp/models
    ```
 
-If you have GPUs available, add the flag `--gpu N` right before `--rm`, where `N` is the number of GPUs you have available.
+If you have GPUs available, you can install the [nvidia-docker](https://github.com/NVIDIA/nvidia-docker) runtime and then add the flag `--gpus all` right before `--rm`.