脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 网站技巧 → 服务器 → 云和虚拟化 → docker → docker 使用GPU

docker 使用GPU的过程详解

 更新时间：2021年11月12日 17:17:52   作者：WGS.  
本文以TensorFlow2.0为例给大家介绍docker 使用GPU的过程详解，文中给大家介绍了基于拉的tf-gpu镜像构建自己的镜像的操作方法，启动镜像检查GPU是否可用的相关知识，感兴趣的朋友跟随小编一起看看吧

以TensorFlow2.0为例

下载tf-gpu

在docker hub里选择要下载的tf版本(注意选带GPU和py3的)

https://hub.docker.com/r/tensorflow/tensorflow/

在这里插入图片描述

如：

docker pull tensorflow/tensorflow:2.0.3-gpu-py3

如果上述下载超时，可以配置清华源。
或者通过如下命令下载：

docker pull docker.mirrors.ustc.edu.cn/tensorflow/tensorflow:2.0.3-gpu-py3

基于拉的tf-gpu镜像构建自己的镜像

如下dockerfile

FROM docker.mirrors.ustc.edu.cn/tensorflow/tensorflow:2.0.3-gpu-py3

RUN ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo "Asia/Shanghai" > /etc/timezone && \
	pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy pandas sklearn scipy matplotlib seaborn pyyaml h5py hdfs

RUN pip install deepctr[gpu] -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
RUN pip install keras==2.3.1 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com


# cd /data/wangguisen/ad_ctr
# docker build -t tf-deepctr:1.0 -f ./dk/Dockerfile_base .

启动镜像检查GPU是否可用

docker run --gpus '"device=0"' \
  --rm -it --name ad_ctr \
  -v /data/wangguisen/ad_ctr:/data/ad_ctr \
  tf-deepctr:1.0

输入 nvidia-smi 如果出现0号显卡的信息说明成功。

在这里插入图片描述

参考自：

https://blog.csdn.net/weixin_35725559/article/details/112268434

https://zhuanlan.zhihu.com/p/83691871

到此这篇关于docker 使用GPU的过程详解的文章就介绍到这了,更多相关docker 使用GPU内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

相关文章

  • docker容器状态转换管理命令实例详解

    docker容器状态转换管理命令实例详解

    Docker容器只是一个运行于宿主操作系统host OS上的应用进程,所以你需要一个镜像来运行它,Docker镜像以进程的方式运行时就叫做Docker容器,这篇文章主要给大家介绍了关于docker容器状态转换管理命令的相关资料,需要的朋友可以参考下
    2022-05-05
  • Ubuntu Docker 源码编译（1.9.1 ）详解

    Ubuntu Docker 源码编译（1.9.1 ）详解

    这篇文章主要介绍了Ubuntu Docker 源码编译（1.9.1 ）详解的相关资料,这里对编译过程进行了详细的分析介绍，需要的朋友可以参考下
    2016-11-11
  • Docker Compose 使用实例详解

    Docker Compose 使用实例详解

    Docker Compose 是用于定义和运行多容器 Docker 应用程序的工具,通过 YAML 文件配置应用程序服务,一键启动所有关联容器,这篇文章主要介绍了Docker Compose 使用实例,需要的朋友可以参考下
    2025-04-04
  • Docker搭建LibreSpeed的实现步骤

    Docker搭建LibreSpeed的实现步骤

    LibreSpeed 是一个轻量级的网络速度测试工具,本文主要介绍了Docker搭建LibreSpeed的实现步骤,具有一定的参考价值,感兴趣的可以了解一下
    2024-04-04
  • docker inspect 操作详解

    docker inspect 操作详解

    docker inspect是docker客户端的原生命令，用于查看docker对象的底层基础信息。包括容器的id、创建时间、运行状态、启动参数、目录挂载、网路配置等等。另外，该命令也可以用来查看docker镜像的信息，这篇文章主要介绍了docker inspect 详解,需要的朋友可以参考下
    2023-01-01
  • harbor修改配置文件后重启操作

    harbor修改配置文件后重启操作

    这篇文章主要介绍了harbor修改配置文件后重启操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-11-11
  • docker 查看容器日志命令的实现

    docker 查看容器日志命令的实现

    这篇文章主要介绍了docker 查看容器日志命令的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12
  • 基于docker搭建selenium分布式环境

    基于docker搭建selenium分布式环境

    这篇文章主要介绍了基于docker搭建selenium分布式环境,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-11-11
  • Docker 端口映射与容器互联的实现

    Docker 端口映射与容器互联的实现

    Docker的端口映射和容器互联是构建和管理微服务架构时重要的概念,本文主要介绍了Docker 端口映射与容器互联的实现,文中通过示例代码介绍的非常详细,需要的朋友们下面随着小编来一起学习学习吧
    2025-01-01
  • Docker没有启动环境变量的问题解决

    Docker没有启动环境变量的问题解决

    在使用Docker容器时,有时候我们会遇到一个问题,就是在容器中没有启动环境变量,本文主要介绍了Docker没有启动环境变量的问题解决,具有一定的参考价值,感兴趣的可以了解一下
    2024-02-02

最新评论