TensorFlow 是 Google 开发的开源机器学习框架,支持多种平台和编程语言。以下是安装步骤:
1. 环境准备
- 安装 Python(推荐 3.7-3.10 版本)
- 确保已安装
pip
或conda
包管理器 - 安装 NVIDIA CUDA Toolkit(如需 GPU 加速)
2. 安装 TensorFlow
方法一:使用 pip 安装
pip install tensorflow
- 会自动安装 CPU 版本(无 GPU 支持)
- 验证安装:
python -c "import tensorflow as tf; print(tf.__version__)"
方法二:安装 GPU 版本
pip install tensorflow-gpu
- 需提前配置 CUDA 和 cuDNN 环境变量
- 详情可参考 TensorFlow GPU 安装文档
3. 验证安装
- 运行以下代码测试是否成功:
import tensorflow as tf print("TensorFlow 安装完成!", tf.__version__)
- 成功输出表示安装正常
4. 常见问题
- 安装失败:检查 Python 版本兼容性
- GPU 不可用:确认 CUDA/cuDNN 版本匹配
- 依赖冲突:使用
--user
参数或虚拟环境
📌 提示:安装完成后,建议学习 TensorFlow 基础教程 开始实践!