docker 19.3以前的版本,需要单独安装nvidia-docker,才可在容器中使用gpu;启动命令
sudo nvidia-docker run -it -e NVIDIA_VISIBLE_DEVICES=all --name 容器名 镜像ID
docker 19.3及以后的版本中,模型集成了nvidia-docker,可直接在容器中使用gpu,启动命令
sudo docker run -it --gpus all --name 容器名 镜像ID
以上是自己踩过的坑总结的,有不足之处请指正。