美文网首页
docker运行gpu

docker运行gpu

作者: V_6619 | 来源:发表于2021-11-02 21:37 被阅读0次

    使用以下命令即可

    docker run -itd --name 容器名字  --gpus all -p 服务器端口号:容器端口号 镜像名字
    docker run -itd --name imgsea  --gpus all -p 6020:80 ubuntu:20.0
    

    下述为在新的服务器上该如何配置docker运行gpu环境

    在nvidia网址Repository configuration | nvidia-container-runtime,获取nvidia-container-runtime的安装源

    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
    

    安装nvidia-container-runtime

    apt-get install nvidia-container-runtime
    

    启动容器时加入gpus选项,all表示gpu全分配

    docker run -itd --name 容器名字  --gpus all -p 服务器端口号:容器端口号 镜像名字
    docker run -itd --name imgsea  --gpus all -p 6020:80 ubuntu:20.04
    

    进入容器后,用以下命令查看显卡

    nvidia-smi
    

    出现以下表示显卡可用


    1635860136(1).png

    相关文章

      网友评论

          本文标题:docker运行gpu

          本文链接:https://www.haomeiwen.com/subject/ovvfzltx.html