文档首页 > > 最佳实践>

CCI使用Nvidia-smi工具

CCI使用Nvidia-smi工具

分享
更新时间:2021/02/10 GMT+08:00

为了支持查看GPU使用情况的场景,需要在镜像中注入nvidia-smi工具,根据购买的专属节点GPU驱动版本选择不同的nvidia-smi二进制文件。

  1. nvidia-smi获取方式。

    该二进制文件可以在nvidia官网,根据CUDA Toolkit版本选择下载对应版本的nvidia驱动包。

    图1 NVIDIA驱动程序下载

    以CUDA 10.1为例,可以下载得到对应版本的驱动包NVIDIA-Linux-x86_64-418.126.run文件,执行如下命令解压缩:

    sh NVIDIA-Linux-x86_64-418.126.run -x

    在解压缩后的目录下即可找到官方提供的nvidia-smi工具二进制文件。

  2. 注入nvidia-smi工具。

    步骤1中得到的nvidia-smi二进制文件放在Dockerfile所在目录下,然后在同目录下的Dockerfile中新增注入语句:

    # inject nvidia-smi tool
    COPY nvidia-smi /usr/bin/
    RUN chmod +x /usr/bin/nvidia-smi
    COPY命令会从Dockerfile文件所在的当前目录下去寻找名为nvidia-smi的文件,然后拷贝到镜像的/usr/bin/目录下。

分享:

    相关文档

    相关产品

文档是否有解决您的问题?

提交成功!非常感谢您的反馈,我们会继续努力做到更好!
反馈提交失败,请稍后再试!

*必选

请至少选择或填写一项反馈信息

字符长度不能超过200

提交反馈 取消

如您有其它疑问,您也可以通过华为云社区问答频道来与我们联系探讨

智能客服提问云社区提问