ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

docker和GPU

2021-09-15 19:34:02  阅读:165  来源: 互联网

标签:Off container nvidia GPU docker runtime


架构图

 

 

安装gpu runtime

https://nvidia.github.io/nvidia-container-runtime/

-----

cat /etc/redhat-release 

CentOS Linux release 7.6.1810 (Core) 

------

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.repo |   sudo tee /etc/yum.repos.d/nvidia-container-runtime.repo

sudo yum install nvidia-container-runtime

-----

cat /etc/docker/daemon.json

{

    "default-runtime": "nvidia",

    "runtimes": {

        "nvidia": {

            "path": "/usr/bin/nvidia-container-runtime",

            "runtimeArgs": []

        }

    },

    "registry-mirrors": ["http://28cf50e2.m.daocloud.io"],

    "storage-driver": "overlay2",

    "storage-opts": [

        "overlay2.override_kernel_check=true"

    ],

    "graph": "/data/data/docker",

    "log-driver": "json-file",

    "log-opts": {

      "max-size": "300m",

      "max-file": "3"

    }

}

---

systemctl daemon-reload

 

下载官方镜像 我司使用基于centos 的gpu 版本

 

https://hub.docker.com/r/nvidia/cuda/tags?page=1&ordering=last_updated

docker pull nvidia/cuda:11.4.1-devel-centos8

docker run -d  --gpus '"device=1,2"'  nvidia/cuda:11.4.1-devel-centos8  sleep 30000000000



nvidia-smi Wed Sep 15 11:25:38 2021 +-----------------------------------------------------------------------------+ | NVIDIA-SMI 418.197.02 Driver Version: 418.197.02 CUDA Version: 11.4 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | |===============================+======================+======================| | 0 Tesla V100S-PCI... Off | 00000000:AF:00.0 Off | 0 | | N/A 70C P0 216W / 250W | 6295MiB / 32480MiB | 100% Default | +-------------------------------+----------------------+----------------------+ | 1 Tesla V100S-PCI... Off | 00000000:D8:00.0 Off | 0 | | N/A 66C P0 207W / 250W | 6295MiB / 32480MiB | 100% Default | +-------------------------------+----------------------+----------------------+ +-----------------------------------------------------------------------------+ | Processes: GPU Memory | | GPU PID Type Process name Usage | |=============================================================================| +-----------------------------------------------------------------------------+

 

标签:Off,container,nvidia,GPU,docker,runtime
来源: https://www.cnblogs.com/leleyao/p/15280429.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有