ocpub/oc9-tensorflow repository overview

⁠tensorflow 2.21.0 on OpenCloudOS 9

⁠基本信息

• 框架版本：v2.21.0
• 基础镜像：opencloudos9-cuda-devel:12.8
• Python 版本：3.11
• CUDA 版本： 12.x 或 更高

⁠构建

docker build -t oc9-tensorflow:2.21.0 .

⁠镜像启动命令

docker run -d --gpus all --name oc9-tensorflow oc9-tensorflow:2.21.0

⁠镜像测试命令

docker run --rm --gpus all oc9-tensorflow:2.21.0 bash test.sh docker run --rm --gpus all oc9-tensorflow:2.21.0 python tensor_smoke_test.py

⁠已知问题

对于如下日志输出，属于正常现象。这些日志通常不是致命错误。
核心原因是：TensorFlow 在 import tensorflow as tf 时，会初始化 CUDA/XLA 相关组件，并注册 cuFFT、cuDNN、cuBLAS 等 GPU 插件工厂。
日志里的意思是：这些插件工厂已经注册过一次，现在又尝试注册，所以打印了“already been registered”。
TensorFlow 官方 GitHub 上有多个相同现象的 issue，日志内容和图片中的告警日志基本一致，
包括 Unable to register cuFFT factory、cuDNN factory、cuBLAS factory、computation placer already registered 等

如下图所示告警日志：