TensorFlow 是由谷歌构建的开源机器学习平台,支持在不同的设备上运行,包括 CPU 和 GPU。本教程将指导您在 CentOS 8 上使用 Python 虚拟环境安装 TensorFlow。
在 CentOS 8 上安装 TensorFlow
在 CentOS 8 上安装 TensorFlow 分为以下步骤:
1. 安装 Python 3 和 pip:
CentOS 8 默认未安装 Python。您需要安装 Python 3 和 pip,运行以下命令:
sudo dnf install python3 python3-pip
2. 创建虚拟环境(可选但推荐):
虚拟环境有助于隔离 TensorFlow 和其依赖项,以免与系统Python包发生冲突。创建并激活虚拟环境:
python3 -m venv tensorflow_env source tensorflow_env/bin/activate
3. 更新 pip:
升级 pip 以确保使用最新版本:
pip install --upgrade pip
4. 安装 TensorFlow:
使用 pip 安装 TensorFlow,包括 CPU 支持:
pip install --upgrade tensorflow
如果您有 NVIDIA GPU 并希望使用 GPU 支持,可以使用以下命令安装 TensorFlow GPU 版本:
pip install --upgrade tensorflow-gpu
5. 验证 TensorFlow 安装:
运行以下命令验证 TensorFlow 安装并打印版本号:
python -c 'import tensorflow as tf; print(tf.__version__)'
此时,您已经成功安装了 TensorFlow。
6. 停用虚拟环境(如果使用了虚拟环境):
使用以下命令停用虚拟环境并返回正常外壳:
deactivate
您现在已经安装了 TensorFlow,可以开始使用这个强大的机器学习框架。
希望这些指导可以帮助您更轻松地安装 TensorFlow。如果您需要任何进一步的帮助,请随时提问。
暂无评论内容