docker runオプションで「–gpus all」 をつけて実行すると以下のエラーが出た。
docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].
ERRO[0000] error waiting for container:
以下を実行することで解決。
sudo apt-get install nvidia-container-runtime
service docker restart
nvidia-container-runtimeが必要だった。
参考
docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]]. ERRO[0000] error waiting for container: context canceledを解決する - Qiita
解決法nvidia-container-runtimeを入れて設定し直すとうまくいく。sudo apt-get install nvidia-container-runtimecurl -s …