Ubuntu 22.04全新环境下TensorFlow GPU版本安装报错求助

问题根源 #

通过Ubuntu默认仓库安装的nvidia-cuda-toolkit，文件分布路径（头文件在/usr/include、库文件在/usr/lib）和TensorFlow配置脚本预期的官方CUDA安装结构（统一在/usr/local/cuda目录下）不匹配，导致路径冲突错误。

修复步骤 #

1. 卸载现有冲突组件 #

先移除通过apt安装的CUDA工具包及相关依赖：

sudo apt purge nvidia-cuda-toolkit
sudo apt autoremove
sudo apt clean

2. 安装官方CUDA Toolkit #

选择TensorFlow支持的CUDA版本（以适配TensorFlow 2.15+的CUDA 12.2为例），通过NVIDIA官方deb仓库安装：

# 添加NVIDIA官方GPG密钥
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

# 更新仓库并安装CUDA 12.2
sudo apt update
sudo apt install cuda-12-2

3. 配置环境变量 #

将CUDA路径添加到系统环境变量中，编辑~/.bashrc文件：

nano ~/.bashrc

在文件末尾添加：

export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存退出后，刷新环境变量：

source ~/.bashrc

4. 验证CUDA安装 #

执行以下命令确认CUDA路径和版本正常：

nvcc --version

输出应显示CUDA 12.2的版本信息，且路径指向/usr/local/cuda-12.2。

5. 重新配置TensorFlow #

回到TensorFlow源码目录，重新运行配置脚本：

cd tensorflow
./configure

此时配置脚本会自动识别到正确的CUDA路径，按提示完成配置即可。

额外建议 #

如果无需自定义编译TensorFlow，推荐直接通过pip安装预编译的GPU版本，步骤更简单：

pip install tensorflow[and-cuda]

这条命令会自动安装适配当前CUDA版本的TensorFlow GPU包，无需手动编译。

内容的提问来源于stack exchange，提问作者John Doe Python