基于docker容器ubuntu16.04安装cuda10.1+cudnn7.6.4

最新推荐文章于 2025-10-10 16:01:33 发布

原创

最新推荐文章于 2025-10-10 16:01:33 发布 · 5.5k 阅读

28 ·

CC 4.0 BY-SA版权

文章标签：

#docker #cuda #ubuntu

本文详细指导了如何在Ubuntu系统上安装Docker、NVIDIA-Docker，并通过脚本拉取镜像，演示了如何在Docker容器中利用GPU运行Ubuntu，包括CUDA和CUDNN的安装过程。

1.宿主机环境

2.安装docker

3.安装nvidia-docker(如果想要在docker容器中调用nvidia驱动必须要安装)

5.1CUDA推荐下载.run可以根据提示安装

1.宿主机环境

系统：ubuntu20.04

GPU驱动：nvidia-driver-418-server

CUDA版本：cuda10.1

CUDNN版本：cudnn7.6.4

2.安装docker

安装命令如下：

curl -fsSL https://get.docker.com | bash -s docker --mirror Aliyun
也可以使用国内 daocloud 一键安装命令：

curl -sSL https://get.daocloud.io/docker | sh

测试
docker run hello-world

3.安装nvidia-docker(如果想要在docker容器中调用nvidia驱动必须要安装)

# If you have nvidia-docker 1.0 installed: we need to remove it and all existing GPU containers
docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge -y nvidia-docker

# Add the package repositories
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

# Install nvidia-docker2 and reload the Docker daemon configuration
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

# Test nvidia-smi with the latest official CUDA image
docker run --runtime=nvidia --rm nvidia/cuda:10.1-base nvidia-smi

安装完成nvidia-docker之后会自动在/etc/docker下创建daemon.json文件，修改daemon.json文件入下：

lu@computer:~$ cat /etc/docker/daemon.json
{
    "runtimes": {
        "nvidia": {
            "path": "nvidia-container-runtime",
            "runtimeArgs": []
        }
    },
   "default-runtime": "nvidia",
   "registry-mirrors": [
	    "https://kfwkfulq.mirror.aliyuncs.com",
	    "https://2lqq34jg.mirror.aliyuncs.com",
	    "https://pee6w651.mirror.aliyuncs.com",
	    "https://registry.docker-cn.com",
	    "https://hub-mirrot.c.163.com"
    ],
    "dns": ["8.8.8.8", "8.8.4.4"]
}
lu@computer:~$

修改完成之后重启docker服务：