Horovod in Docker

最新推荐文章于 2023-03-16 22:09:36 发布

a flying bird

最新推荐文章于 2023-03-16 22:09:36 发布

阅读量172

点赞数

分类专栏：分布式深度学习文章标签： docker html java

本文链接：https://blog.csdn.net/m0_37870649/article/details/120477747

版权

本文介绍了如何在Docker环境中安装和运行Horovod。通过`docker run`命令设置容器，并映射端口和挂载目录。尽管在导入Tensorflow时出现GPU警告，但不影响容器运行。Dockerfile中包含了保持容器运行的关键指令，如更新包源、安装Nginx，并以非守护进程模式启动。参考链接指向OSChina上的个人空间。

摘要由CSDN通过智能技术生成

docker run -d -p=3310:8080 -v /home:/home --name mydocker horovod:latest

如上安装成功后，导入tensorflow时有提示waring,忽略。提示的是GPU的错误。

Dockerfile文件如下：

FROM ubuntu:18.04

# TensorFlow version is tightly coupled to CUDA and cuDNN so it should be selected carefully
ARG TENSORFLOW_VERSION=2.5.0
ARG PYTORCH_VERSION=1.8.1
ARG PYTORCH_LIGHTNING_VERSION=1.2.9
ARG TORCHVISION_VERSION=0.9.1
ARG MXNET_VERSION=1.8.0.post0

ARG PYSPARK_PACKAGE=pyspark==3.1.1
ARG SPARK_PACKAGE=spark-3.1.1/spark-3.1.1-bin-hadoop2.7.tgz

# Python 3.7 is supported by Ubuntu Bionic out of the box
ARG PYTHON_VERSION=3.7

# Set default shell to /bin