rocm/tensorflow Docker 镜像 - 轩辕镜像 | Docker 镜像高效稳定拉取服务

概述

本仓库托管支持ROCm后端的Tensorflow发布版Docker镜像。

支持的GPU型号

镜像可在以下GPU型号上运行：

• gfx900（Vega10-type GPU - MI25, Vega56, Vega64）
• gfx906（Vega20-type GPU - MI50, MI60）
• gfx908（MI100）
• gfx90a（MI200）
• gfx1030（Navi21）

环境配置

如需配置ROCm Docker环境，请参考以下安装指南： [***]

镜像特性

• 带有-dev标签的容器包含从源码重建tensorflow-rocm所需的所有依赖项。

使用方法（部署示例）

推荐部署命令

以下是启动TF2.5镜像的推荐命令：

1. 创建drun别名（包含必要的容器运行参数）： alias drun='sudo docker run -it --network=host --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size 16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $HOME/dockerx:/dockerx'

2. 使用drun运行镜像： drun rocm/tensorflow:latest

说明：--device=/dev/kfd和--device=/dev/dri用于启用GPU设备访问；--shm-size 16G设置共享内存大小；-v $HOME/dockerx:/dockerx将主机目录挂载到容器内。

镜像基础信息

• ROCm 3.1及之后版本的镜像基于Ubuntu 18.04。
• 更早版本的镜像基于Ubuntu 16.04，除非镜像标签中另有说明。