如何安装tensorflow

发布时间：2026-09-24 12:44:29 来源：亿速云阅读：105 作者：小樊栏目：软件技术

下面分Windows / macOS / Linux 常见情况说明如何安装 TensorFlow（以 TensorFlow 2.x 为准，目前主流）。

一、安装前准备

1. 安装 Python

2. 安装 pip

python -m ensurepip --upgrade

二、使用虚拟环境（强烈推荐）

创建虚拟环境

python -m venv tf_env

激活环境

Windows：

tf_env\Scripts\activate

macOS / Linux：

source tf_env/bin/activate

三、安装 TensorFlow

1. CPU 版本（最简单）

pip install --upgrade pip
pip install tensorflow

验证：

python -c "import tensorflow as tf; print(tf.__version__)"

2. GPU 版本（可选，需要 NVIDIA 显卡）

要求

NVIDIA GPU（Compute Capability ≥ 3.5）
安装 CUDA 和 cuDNN（版本需匹配 TensorFlow）
安装 NVIDIA 驱动

安装

pip install tensorflow

TensorFlow 2.x 已默认支持 GPU（无需单独安装 tensorflow-gpu）

验证 GPU：

import tensorflow as tf
print(tf.config.list_physical_devices('GPU'))

四、常见问题

1. pip 版本太旧

pip install --upgrade pip

2. 网络慢

pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple

3. Apple Mac（M1/M2/M3）

pip install tensorflow-macos
pip install tensorflow-metal

五、最小测试代码

import tensorflow as tf

a = tf.constant([1, 2, 3])
b = tf.constant([4, 5, 6])
print(a + b)

如果你告诉我：

操作系统
是否用 GPU
Python 版本

我可以给你更精确的命令。

向AI问一下细节