zoukankan      html  css  js  c++  java
  • ubuntu 在docker中使用gpu,安装nvidiacontainerruntime

    1. 参考https://nvidia.github.io/nvidia-container-runtime/ 配置nvidia 源

    主要执行:

    curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
      sudo apt-key add -
    distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
    curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
      sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
    sudo apt-get update

    2. 参考https://gitee.com/wuxler/nvidia-container-runtime?_from=gitee_search 安装nvidia-container-runtime

    主要执行:

    安装运行时

    sudo apt-get install nvidia-container-runtime

    停止docker

    systemctl stop docker
    把运行时添加到docker中:
    dockerd --add-runtime=nvidia=/usr/bin/nvidia-container-runtime

    创建一个容器:

    创建容器时一定要指定gpus参数

    docker run -it --gpus=all 镜像名  执行命令

    进入容器后执行

    nvidia-smi

    如果正常会有GPU信息

    PS: 创建容器时偌提示:

    docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].
    ERRO[0000] error waiting for container: context canceled
    apt-get install nvidia-container-toolkit
    systemctl restart docker
  • 相关阅读:
    函数式编程
    JSONP
    用javascript实现base64编码器
    图片Ping
    CORS
    深入理解ajax系列第五篇——进度事件
    文件File
    深入理解ajax系列第四篇——FormData
    Blob
    深入理解ajax系列第三篇——响应解码
  • 原文地址:https://www.cnblogs.com/GengMingYan/p/14962878.html
Copyright © 2011-2022 走看看