脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 网站技巧 → 服务器 → 云和虚拟化 → 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设置代理的方法教程，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2017-08-08
  • Docker如何使用nginx搭建tomcat集群(图文详解)

    Docker如何使用nginx搭建tomcat集群(图文详解)

    这篇文章主要介绍了Docker使用nginx搭建tomcat集群的教程,本文图文并茂给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下
    2019-12-12
  • Docker consul的容器服务更新与发现的问题小结

    Docker consul的容器服务更新与发现的问题小结

    这篇文章主要介绍了Docker consul的容器服务更新与发现,讲解了服务注册与发现的基本概念讲解，本文给大家介绍的非常详细，需要的朋友可以参考下
    2021-08-08
  • Docker部署前后端分离项目的实现示例

    Docker部署前后端分离项目的实现示例

    本文主要介绍了Docker部署前后端分离项目的实现示例，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2021-11-11
  • 基于docker安装tensorflow的完整步骤

    基于docker安装tensorflow的完整步骤

    TensorFlow 随着AlphaGo的胜利也火了起来。 下面这篇文章主要给大家介绍了关于基于docker安装tensorflow的相关资料，文中通过示例代码介绍的非常详细，需要的朋友们下面随着小编来一起学习学习吧。
    2018-02-02
  • Docker-compose的安装和设定详细步骤

    Docker-compose的安装和设定详细步骤

    这篇文章主要介绍了Docker-compose的安装和设定的相关资料,需要的朋友可以参考下
    2017-03-03
  • Docker MQTT安装使用教程

    Docker MQTT安装使用教程

    今天小编就为大家分享一篇关于Docker MQTT安装使用教程，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧
    2018-10-10
  • CentOS8下的Docker使用详解

    CentOS8下的Docker使用详解

    这篇文章主要介绍了CentOS8下的Docker使用,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2020-11-11
  • docker-mysql连接方式

    docker-mysql连接方式

    这篇文章主要介绍了docker-mysql连接方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
    2023-05-05
  • docker安装nginx之后无法访问的解决

    docker安装nginx之后无法访问的解决

    这篇文章主要介绍了docker安装nginx之后无法访问的解决方案,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-09-09

最新评论