TensorFlow 是一个开源机器学习框架,支持多种平台和语言。以下是在 Linux 系统上安装 TensorFlow 的常用方法:
1. 使用 pip 安装 📦
pip install tensorflow
📌 注意:确保已安装 Python 3.7+ 和 pip。如果遇到权限问题,可添加
--user
参数。
2. 从源码编译 🧰
- 克隆 TensorFlow 仓库:
git clone https://github.com/tensorflow/tensorflow.git
- 进入目录并配置构建环境:
cd tensorflow ./configure
- 编译并安装:
bazel build --config=opt //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
3. 使用 Docker 🐳
运行官方镜像:
docker run -it --rm tensorflow/tensorflow:latest
或自定义环境:
docker run -it --gpus=yes -v $(pwd):/workspace tensorflow/tensorflow:latest-gpu
4. 验证安装 ✅
执行以下代码检查版本:
import tensorflow as tf
print(tf.__version__)
若输出版本号,说明安装成功!
如需深入了解 TensorFlow 的系统要求或配置选项,可访问 TensorFlow 官方文档 进行扩展阅读。