gpt4 book ai didi

docker 使用GPU的过程详解

转载 作者:qq735679552 更新时间:2022-09-27 22:32:09 26 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章docker 使用GPU的过程详解由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

以TensorFlow2.0为例 。

下载tf-gpu

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

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

docker 使用GPU的过程详解

如:

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 。

?
1
2
3
4
5
6
7
8
9
10
11
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是否可用

?
1
2
3
4
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号显卡的信息说明成功.

docker 使用GPU的过程详解

参考自:

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

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

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

原文链接:https://blog.csdn.net/qq_42363032/article/details/121268251 。

最后此篇关于docker 使用GPU的过程详解的文章就讲到这里了,如果你想了解更多关于docker 使用GPU的过程详解的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com