docker中使用英伟达驱动

安装以下依赖并重新启动：

sudo apt install -y nvidia-docker2
sudo systemctl daemon-reload
sudo systemctl restart docker

使用类似如下命令创建容器：

sudo docker run -it --name gpu-env --gpus all ubuntu

注意加上--gpus all这样的参数即可