TensorFlow 是一个开源机器学习框架,支持多种平台和硬件配置。以下是详细的安装步骤,帮助您快速开始:
1. 确定环境需求 📋
操作系统:支持 Windows、Linux、macOS
Python 版本:推荐使用 Python 3.8-3.11
硬件要求:
- CPU 安装:无需额外配置
- GPU 加速:需安装 NVIDIA 驱动和 CUDA 工具包(推荐参考 GPU 配置指南)
2. 安装步骤 🛠
2.1 通过 pip 安装
pip install tensorflow
⚠️ 若需 GPU 支持,请安装 TensorFlow GPU 版本。
2.2 通过 Docker 安装
docker run -it --rm gcr.io/tensorflow/tensorflow:latest
🌐 更多 Docker 配置详情请访问 Docker 部署文档。
2.3 从源码编译(高级用户)
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
./configure
bazel build --config=opt --config=cuda //tensorflow:libtensorflow.so
📚 源码编译教程可参考 官方开发者指南。
3. 验证安装 ✅
运行以下代码检查 TensorFlow 是否正常:
import tensorflow as tf
print(tf.__version__)
🎉 若输出版本号,说明安装成功!如需进一步测试,可访问 TensorFlow 官方示例。
4. 常见问题解答 ❓
Q: 安装后无法使用 GPU?
A: 请确保已正确安装 CUDA 和 cuDNN,参考 GPU 配置指南。Q: 如何升级 TensorFlow?
A: 使用pip install --upgrade tensorflow
即可。
📦 更多 TensorFlow 资源 | 📘 官方文档 | 🛠 安装工具包下载