TensorFlow 是一个强大的机器学习框架,以下是几种常见的安装方式:
1. 使用 pip 安装 📦
pip install tensorflow
✅ 适用于大多数 Python 环境,支持 CPU 和 GPU 版本。
📌 注意:确保已安装 Python 3.7-3.11,并通过 Python 官方文档 确认版本兼容性。
2. 从源码编译 🧑💻
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
./configure
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
pip install /tmp/tensorflow_pkg/tensorflow-*.whl
🔍 适合需要自定义配置的用户,详细步骤可参考 TensorFlow 源码编译教程。
3. 使用 Docker 🐳
docker run --rm -it tensorflow/tensorflow:latest
🌐 一键启动预配置环境,适合快速测试。更多 Docker 镜像信息请查看 Docker 官方仓库。
4. 其他方式
- Colab 环境:直接在 Google Colab 中使用,无需本地安装
📝 Colab 安装指南 - Anaconda:通过
conda install -c conda-forge tensorflow
安装
📦 Anaconda 安装说明
如需验证安装是否成功,可运行以下代码:
import tensorflow as tf
print(tf.__version__)
✅ 若输出版本号,则表示安装成功!
返回 TensorFlow 主页 了解更多使用技巧。