docker使用GPU(nvidia-docker)

docker使用GPU(nvidia-docker)在docker19以前都需要单独下载nvidia-docker1或nvidia-docker2来启动容器,自从升级了docker19后跑需要gpu的docker只需要加个参数–gpusall即可为了让docker支持nvidia显卡,英伟达公司开发了nvidia-docker。该软件是对docker的包装,使得容器能够看到并使用宿主机的nvidia显卡。为什么我的本机CUDA版本是10.0而docker中是cuda9.0却也能够跑起来一……

      在docker19以前都需要单独下载nvidia-docker1或nvidia-docker2来启动容器,自从升级了docker19后跑需要gpu的docker只需要加个参数–gpus all 即可

      为了让docker支持nvidia显卡,英伟达公司开发了nvidia-docker。该软件是对docker的包装,使得容器能够看到并使用宿主机的nvidia显卡。

      为什么我的本机CUDA版本是10.0 而docker中是cuda9.0却也能够跑起来

      一般来说,一台机器只能有一个版本的驱动(nvidia-smi中显示的Driver Version),然而CUDA是可以多版本共存的,使用docker镜像或anaconda虚拟环境可以隔开不同的版本

所以,我本机的Driver Version是430.64

docker使用GPU(nvidia-docker)

 理论上说10.1及以下的cuda都可以装

# 使用所有GPU
$ docker run --gpus all nvidia/cuda:9.0-base nvidia-smi

# 使用两个GPU
$ docker run --gpus 2 nvidia/cuda:9.0-base nvidia-smi

# 指定GPU运行
$ docker run --gpus '"device=1,2"' nvidia/cuda:9.0-base nvidia-smi
$ docker run --gpus '"device=UUID-ABCDEF,1"' nvidia/cuda:9.0-base nvidia-smi

这里nvidia/cuda:9.0-base是nvidia官方提供的docker镜像

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
如需转载请保留出处:https://bianchenghao.cn/34524.html

(0)
编程小号编程小号

相关推荐

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注