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

星枫大大_8571

星枫大大_8571

2026-05-27

684人浏览

原创

tf.config.list_physical_devices('gpu') 返回空列表,根本原因在于cuda/cudnn、nvidia驱动、python及tensorflow版本未严格对齐;需先通过nvidia-smi确认驱动支持的最高cuda版本,再按官方兼容表精确匹配安装对应cuda toolkit与cudnn,并使用干净python 3.9环境pip install tensorflow(非tensorflow-gpu),最后验证动态库路径与版本一致性。

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

tf.config.list_physical_devices('GPU') 返回空列表,不是因为没装 TensorFlow,而是底层 CUDA/cuDNN 与驱动、Python 版本之间有一环没对齐。直接 pip install tensorflow 装出来的几乎肯定是 CPU 版——想用 GPU,必须手动控链。

先确认显卡驱动和 CUDA 支持上限

别急着装 CUDA,先看驱动允许你用什么版本:nvidia-smi 右上角显示的 “CUDA Version: 12.x” 是**驱动支持的最高版本**,不是你已装的 CUDA。这个数字决定了你能往下选哪个 CUDA Toolkit。

  • 如果显示 CUDA Version: 12.4,说明驱动 ≥550,可装 CUDA 11.8 或 12.2(取决于你要用的 TensorFlow 版本)
  • 如果显示 CUDA Version: 11.6,那 CUDA 12.x 就不能装,否则 libcudart.so.12 找不到
  • nvcc -Vnvidia-smi 输出不一致完全正常;前者查的是已安装的 Toolkit,后者查的是驱动能力

创建 Python 3.9 环境并装 tensorflow(不是 tensorflow-gpu)

TensorFlow 2.10+ 已废弃 tensorflow-gpu 包名,统一用 tensorflow,GPU 支持由底层 CUDA 自动触发。但前提是环境干净、版本对得上。

onewo-rtlinux
onewo-rtlinux

Linux 实时编程助手。生成、审查并修改用于周期性控制任务和中断驱动程序的 C 代码。强制实施实时调度策略。

下载
  • 用 conda 创建隔离环境:conda create -n tf-gpu python=3.9 —— Python 3.9 是目前 wheel 兼容率最高、cuDNN 自动匹配最稳的基线
  • 激活后优先用 pip:conda activate tf-gpupip install tensorflow(conda 渠道的 tensorflow 包有时会漏掉 CUDA 动态库绑定)
  • 别用 conda install tensorflow-gpu,该包名在 TF 2.10+ 中已被移除,强行装会降级或冲突

验证 GPU 是否真正可用,不止看 import 成功

import 不报错 ≠ GPU 可用。import tensorflow as tf 成功只说明框架加载了,关键要看设备发现逻辑是否走通。

  • 运行:python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
  • 返回 [PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')] 才算成功
  • 若返回空列表,常见原因:CUDA 路径没进 LD_LIBRARY_PATHlibcudnn.so.8 找不到、或实际装的 CUDA 版本与 TensorFlow 编译时所用版本不符(例如 TF 2.15 要求 CUDA 12.2,你装了 12.4 就不行)

conda init 没选 yes 就等于白装 Anaconda

执行 bash Anaconda3-*.sh 后,出现 Do you wish the installer to initialize Anaconda3 by running conda init? [yes|no] 提示时按回车(默认 no),后果是:conda 命令在新终端里根本不可用。

  • 补救只有两个办法:重跑安装脚本并输 yes;或手动执行 ~/anaconda3/bin/conda init bashsource ~/.bashrc
  • source ~/anaconda3/etc/profile.d/conda.sh 只对当前终端生效,关掉就失效——这不是临时方案,是误操作
  • 验证是否生效:which conda 应输出类似 /home/xxx/anaconda3/bin/conda,且新终端里 conda --version 有输出
CUDA 和 cuDNN 的版本号必须精确匹配 TensorFlow 官方构建表,差一个小数点都可能让 tf.config.list_physical_devices('GPU') 静默失败。很多人卡在最后一步,不是不会装,是没意识到“装上了”和“能调用”之间隔着三道动态链接检查。

相关文章

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

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

下载

相关标签:

linux linux配置

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系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

626

16

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

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

2026.02.10

176

19

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

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

2023.06.21

3610

4

如何安装LINUX
如何安装LINUX

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

2023.06.29

3133

10

linux find
linux find

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

2023.06.30

2193

5

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

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

2023.07.05

5210

7

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

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

2023.07.06

4882

7

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

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

2023.07.20

2933

8

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Valgrind支持平台说明
Valgrind支持平台说明

共0课时 | 0人学习

CLion 远程开发实战
CLion 远程开发实战

共0课时 | 0人学习