docker 容器GPU环境搭建

文章目录

  • docker install
    • 准备工作
    • 安装Docker CE
    • 修改Docker默认储存位置
  • 安装nvidia-docker2 容器GPU化关键
    • 运行nvidia-docker
    • 官方指定教程

docker install

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无法共享,很可能本地机器的不一致导致每台机器都需要去重

你可能感兴趣的:(机器学习原理讲解与代码实现,linux入门到精通,大数据与云计算入门到实战,深度学习,强化学习,python小工具编写)