Nvidia-Docker安装使用 -- 可使用GPU的Docker容器
nvidia-docker 是一个可以使用 GPU 的 docker , nvidia-docker 是在 docker 上做了一层封装,通过 nvidia-docker-plugin ,然后调用到 docker 上,其最终实现的还是在 docker 的启动命令上携带一些必要的参数。因此在安装 nvidia-docker 之前,还是需要安装 docker 的。 docker 一般都是使用基于 CPU 的应用,而如果是 GPU 的话,就需要安装特有的硬件环境,比如需要安装 nvidia driver 。所以 docker 容器并不直接支持 Nvidia GPU 。为了解决这个问题,最早的处理办法是在容器内部,全部重新安装 nvidia driver ,然后通过设置相应的设备参数来启动 container ,然而这种办法是很脆弱的。因为宿主机的 driver 的版本必须完全匹配容器内的 driver 版本,这样导致 docker image 无法共享,很可能本地机器的不一致导致每台机器都需要去重复操作,这很大的违背了 docker 的设计之初。 为了使 docker image 能很便利的使用 Nvidia GPU ,从而产生了 nvidia-docker ,由它来制作 nvidia driver 的 image ,这就要求在目标机器上启动 container 时