如何在Linux中安装TensorFlow Linux配置深度学习环境的方法

风芳大大_1775

风芳大大_1775

2026-06-02

1013人浏览

原创

tf.config.list_physical_devices('gpu') 返回空列表,表明gpu未被tensorflow识别,根本原因是nvidia驱动、cuda toolkit、cudnn、python四者版本未对齐;需依次验证nvidia-smi、nvcc -v、libcudnn.so.8存在性及ld_library_path配置。

如何在linux中安装tensorflow linux配置深度学习环境的方法

tf.config.list_physical_devices('GPU') 返回空列表,不是因为没装 TensorFlow,而是驱动、CUDA、cuDNN、Python 版本四者中至少一个没对齐。别急着重装,先查清楚哪一环断了。

确认 NVIDIA 驱动是否真就绪

nvidia-smi 能显示显卡信息 ≠ 驱动可被 CUDA 调用。常见假成功现象: - nvidia-smi 正常输出,但 nvcc -V 报 command not found → CUDA Toolkit 根本没装 - nvidia-smi 右上角写 “CUDA Version: 12.4”,但你装的是 CUDA 12.3 → 没问题,这只是驱动支持上限 - nvidia-smi 报 “NVIDIA-SMI has failed…” → 驱动未安装或与内核不兼容,必须重装对应版本的 nvidia-driver(如 Ubuntu 22.04 推荐 nvidia-driver-535)

选对 Python 和 TensorFlow 组合,别碰 3.11 或 3.8

TensorFlow 官方支持 Python 3.8–3.11,但实操中: - Python 3.8:部分新语法(如 typing.Union 简写)缺失,某些 tensorflow-probability 包会 import 失败 - Python 3.11:很多预编译 wheel 还没发布,pip install tensorflow 可能退回到源码编译,失败率高 - Python 3.9:conda 渠道默认打包版本,cudatoolkit 和 cudnn 自动匹配成功率最高

所以直接执行:
conda create -n tf-gpu python=3.9
conda activate tf-gpu

装 tensorflow 不要加 -gpu 后缀,也别用 conda install tensorflow-gpu

从 TensorFlow 2.10 开始,tensorflow-gpu 包名已废弃。现在统一用: pip install tensorflow

但前提是:

TensorFlow Linux版 2.18.1
TensorFlow Linux版 2.18.1

TensorFlow 2.18.1 历史版本下载,来自 PyPI 官方发布,适合旧项目兼容、实验复现和指定环境安装。

下载
  • 环境里已有匹配的 cudatoolkit 和 cudnn(conda 会自动装,pip 不会)
  • 或你已手动把 libcudart.so.XX 和 libcudnn.so.8 所在路径加进 LD_LIBRARY_PATH

更稳的做法是:
conda install cudatoolkit=11.8 cudnn=8.6(对应 TF 2.12)
pip install tensorflow==2.12.0

验证 GPU 是否真可用,别只看 import 成功

import tensorflow as tf 成功不代表 GPU 就绪。必须跑这句: print(tf.config.list_physical_devices('GPU'))

如果输出 [],按顺序排查:

  • ls /usr/local/cuda-11.8/lib64/libcudnn.so.8 是否存在
  • echo $LD_LIBRARY_PATH 是否包含该路径(如没有,补 export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH)
  • python -c "import ctypes; print(ctypes.CDLL('libcudnn.so.8'))" 是否报 FileNotFoundError

最常被忽略的是:装完 CUDA 后没重启终端,或没把 source /usr/local/cuda-11.8/bin/setup.sh 加进 ~/.bashrc。路径漏了,tensorflow 就只能降级用 CPU。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习
Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习

PyTorch 是一种用于构建深度学习模型的功能完备框架,是一种通常用于图像识别和语言处理等应用程序的机器学习。 使用Python 编写,因此对于大多数机器学习开发者而言,学习和使用起来相对简单。 PyTorch 的独特之处在于,它完全支持GPU,并且使用反向模式自动微分技术,因此可以动态修改计算图形。

2025.12.22

68

5

Python 深度学习框架与TensorFlow入门
Python 深度学习框架与TensorFlow入门

本专题深入讲解 Python 在深度学习与人工智能领域的应用,包括使用 TensorFlow 搭建神经网络模型、卷积神经网络(CNN)、循环神经网络(RNN)、数据预处理、模型优化与训练技巧。通过实战项目(如图像识别与文本生成),帮助学习者掌握 如何使用 TensorFlow 开发高效的深度学习模型,并将其应用于实际的 AI 问题中。

2026.01.07

686

16

TensorFlow2深度学习模型实战与优化
TensorFlow2深度学习模型实战与优化

本专题面向 AI 与数据科学开发者,系统讲解 TensorFlow 2 框架下深度学习模型的构建、训练、调优与部署。内容包括神经网络基础、卷积神经网络、循环神经网络、优化算法及模型性能提升技巧。通过实战项目演示,帮助开发者掌握从模型设计到上线的完整流程。

2026.02.10

176

19

磁盘配额是什么
磁盘配额是什么

磁盘配额是计算机中指定磁盘的储存限制,就是管理员可以为用户所能使用的磁盘空间进行配额限制,每一用户只能使用最大配额范围内的磁盘空间。php中文网为大家提供各种磁盘配额相关的内容,教程,供大家免费下载安装。

2023.06.21

3910

4

如何安装LINUX
如何安装LINUX

本站专题提供如何安装LINUX的相关教程文章,还有相关的下载、课程,大家可以免费体验。

2023.06.29

3353

10

linux find
linux find

find是linux命令,它将档案系统内符合 expression 的档案列出来。可以指要档案的名称、类别、时间、大小、权限等不同资讯的组合,只有完全相符的才会被列出来。find根据下列规则判断 path 和 expression,在命令列上第一个 - ( ) , ! 之前的部分为 path,之后的是 expression。还有指DOS 命令 find,Excel 函数 find等。本站专题提供linux find相关教程文章,还有相关

2023.06.30

2493

5

linux修改文件名
linux修改文件名

本专题为大家提供linux修改文件名相关的文章,这些文章可以帮助用户快速轻松地完成文件名的修改工作,大家可以免费体验。

2023.07.05

6030

7

linux系统安装教程
linux系统安装教程

linux系统是一种可以免费使用,自由传播,多用户、多任务、多线程、多CPU的操作系统。本专题提供linux系统安装教程相关的文章,大家可以免费体验。

2023.07.06

5662

7

linux查看文件夹大小
linux查看文件夹大小

Linux是一种自由和开放源码的类Unix操作系统,存在着许多不同的Linux版本,但它们都使用了Linux内核。Linux可安装在各种计算机硬件设备中,比如手机、平板电脑、路由器、视频游戏控制台、台式计算机、大型机和超级计算机。linux怎么查看文件夹大小呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.07.20

3333

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程