记录在docker使用宿主机显卡驱动(ubuntu18.04)

1. 宿主机安装好显卡驱动、cuda、cudnn

2. 安装docker

使用官方安装脚本

curl -fsSL https://get.docker.com | bash -s docker --mirror Aliyun

3. 安装nvidia-docker

curl -fsSL https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -fsSL https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo