Run Script Should Check for nvidia-container-runtime
Description
nvidia-docker
/ --runtime=nvidia
has been replaced with the --gpus
flag on recent (19.03+) versions of Docker. The run.sh
script currently assumes that --runtime=nvidia
is the only way to enable NVIDIA support.
Implementation considerations
The script should check for installation of the package nvidia-container-runtime
. One way this can be done is by checking for the return code of the following:
apt search nvidia-container-runtime | grep installed
This line returning a 0 would indicate that it is installed while any other return value would indicate that it is not.
Alternatives
None that I know of.