Notify user to install nvidia-container-runtime

[lwshang]

The package nvidia-container-runtime is necessary to run docker with Nvidia GPU. I followed the instruction in README and tried tensorman run --gpu bash. It failed and showed following message:

could not select device driver "" with capabilities: [[gpu]]

I searched this error message and the solution was just install the dependency nvidia-container-runtime.

I believe it will be better to give some useful information for a user like me who try to run docker with GPU for the first time.

Ideally, if --gpu is set and nvidia-container-runtime has not been installed, tensorman may friendly tell the user to install the dependent packages by directly showing the apt-get install command.

[B3njimin]

I am having a similar issue only if I try to install the nvidia-container-runtime I get this message

Package nvidia-container-runtime is not available, but is referred to by another package. This may mean that the package is missing, has been obsoleted, or is only available from another source However the following packages replace it: nvidia-docker2 nvidia-container-toolkit nvidia-container-toolkit-base

[mmstick]

You should install nvidia-docker2

[B3njimin]

thank you I also had to run sudo systemctl restart docker in order for it to work.