环境要求
本文档以openEuler 24.03 LTS SP3 + TensorFlow 2.21.0版本为例进行GPU安装验证。除此之外,本章节介绍了鲲鹏服务器上安装TensorFlow可支持的其他环境信息。
可支持的版本
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用户态库。
依赖组件要求
组件 |
版本 |
|---|---|
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 |
父主题: 安装前准备