目录下载tf-gpu基于拉的tf-gpu镜像构建自己的镜像启动镜像检查GPU是否可用以Tensorflow2.0为例 下载tf-gpu 在Docker hub里选择要下载的tf版本(
以Tensorflow2.0为例
在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
如下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 .
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内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！
--结束END--
本文标题: docker 使用GPU的过程详解
本文链接: https://www.lsjlt.com/news/157217.html(转载时请注明来源链接)
有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
下载Word文档到电脑，方便收藏和打印～
2024-05-13
2024-05-13
2024-05-11
2024-05-11
2024-05-10
2024-05-07
2024-04-30
2024-04-30
2024-04-30
2024-04-29
回答
回答
回答
回答
回答
回答
回答
回答
回答
回答
官方手机版
微信公众号
商务合作
0