diff --git a/README.md b/README.md index 32dfaafeb0c24a0ee697ec1d8a48b76d815270d9..bcc75aa10b40246c07021d431452703e414619d8 100644 --- a/README.md +++ b/README.md @@ -70,6 +70,9 @@ the script `run_command_in_container.sh`. If you execute ``` it will invoke the Python interpreter that has been installed into the container image. You will see which python version has been installed into the container. Instead, you can, of course, execute any other command and pass arbitrary arguments. +A note on GPUs. If you want to be able to use GPUs inside of the container, you must use the [NVIDIA Container runtime](https://developer.nvidia.com/nvidia-container-runtime) and +add the flag `--runtime=nvidia` to the all calls to `docker run`. + # Export the docker container Before you start into this step, ssh to the supercomputer and also clone the repository there. ```bash