您当前的位置:首页 > 网站建设 > 网站维护
| php | asp | css | H5 | javascript | Mysql | Dreamweaver | Delphi | 网站维护 | 帝国cms | React | 考试系统 | ajax | jQuery |

docker 使用GPU的过程详解

51自学网 2022-07-04 11:29:33
  网站维护

以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-py3RUN 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 hdfsRUN pip install deepctr[gpu] -i http://pypi.douban.com/simple --trusted-host pypi.douban.comRUN 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


下载地址:
SpringBoot 整合 Docker的详细过程
docker-compose搭建mongodb、mysql的详细过程
51自学网,即我要自学网,自学EXCEL、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。
京ICP备13026421号-1