最初的docker是不支持gpu的;
为了让docker支持nvidia显卡,英伟达公司开发了nvidia-docker。
该软件是对docker的包装,使得容器能够看到并使用宿主机的nvidia显卡。
根据网上的资料,从docker 19版本之后,nvidia-docker成为了过去式。不需要单独去下nvidia-docker这个独立的docker应用程序,也就是说gpu docker所需要的Runtime被集成进docker中,使用的时候用--gpus参数来控制。
前置环境:
需要在本机上安装nvidia驱动,cuda和cudnn。
可参考:
Ubuntu手动安装Nvidia显卡驱动的方法
Ubuntu1804安装cuda和cudnn
官网地址:
https://docs.nvidia.com/ai-enterprise/deployment-guide/dg-docker.html
step1.Add the package repositories
添加包存储库,在终端依次输入以下命令:
$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
$ curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
step2.下载安装nvidia-container-toolkit包
Download information from all configured sources about the latest versions of the packages and install the nvidia-container-toolkit package:
$ sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
step3.重启docker服务
Restart the Docker service
$ sudo systemctl restart docker
step4.测试及使用方法
#### Test nvidia-smi with the latest official CUDA image
$ sudo docker run --gpus all nvidia/cuda:11.0-base nvidia-smi
使用docker run进入容器的时候加入--gpus参数。
此时在容器内就可以使用nvidia-smi命令,查询nvidia是否可用。
《完》
点击上方小卡片关注我
上一篇
docker去掉sudo权限的方法