开发者
资源
我要评分
获取效率
正确性
完整性
易理解
在线提单
论坛求助

环境要求

本文档以openEuler 24.03 LTS SP3 + TensorFlow 2.21.0版本为例进行GPU安装验证。除此之外,本章节介绍了鲲鹏服务器上安装TensorFlow可支持的其他环境信息。

可支持的版本

表1 版本范围

OS版本

Python版本

支持的TensorFlow版本范围

openEuler 24.03 LTS SP3

Python 3.11.6

TensorFlow 2.21.0

openEuler 22.03 LTS SP4

Python 3.9.9

不适用于TensorFlow 2.21.0;TensorFlow 2.21.0要求更新的Python版本

TensorFlow 2.21.0的官方pip GPU安装路径使用CUDA 12系列NVIDIA pip包。源码构建GPU版本时,应使用与构建配置匹配的CUDA Toolkit、cuDNN和NVIDIA驱动。验证环境中NVIDIA驱动版本为580.159.04,驱动侧显示CUDA 13.0,驱动能够兼容运行CUDA 12用户态库。

依赖组件要求

表2 依赖组件及版本

组件

版本

python3

3.11.6-20.oe2403sp3

pip

26.1.1

TensorFlow

2.21.0

CUDA用户态依赖

CUDA 12系列的NVIDIA pip包

NVIDIA Driver

580.159.04

GPU

NVIDIA A100-PCIE-40GB