TensorFlow 是一个开源的深度学习框架，广泛用于各种机器学习和深度学习任务。在 Linux 系统下安装和配置 TensorFlow 是使用 TensorFlow 进行研究和开发的重要步骤。

安装 TensorFlow

系统要求

  • Ubuntu 16.04 或更高版本
  • CentOS 7 或更高版本
  • Debian 9 或更高版本

安装步骤

  1. 更新系统包列表：
sudo apt-get update
  1. 安装依赖项：
sudo apt-get install -y python3-pip python3-dev
  1. 创建虚拟环境：
python3 -m venv tensorflow_env
source tensorflow_env/bin/activate
  1. 安装 TensorFlow：
pip install tensorflow
  1. 验证安装：
python -c "import tensorflow as tf; print(tf.__version__)"

配置 TensorFlow

GPU 支持

如果您使用的是具有 NVIDIA GPU 的系统，您需要安装 CUDA 和 cuDNN。

  1. 安装 CUDA：
sudo apt-get install -y nvidia-cuda-toolkit
  1. 安装 cuDNN：

从 NVIDIA 官网下载 cuDNN，解压后将其内容复制到 /usr/local/cuda/lib64 和 /usr/local/cuda/include 目录。

  1. 配置环境变量：
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
  1. 安装 cuDNN 兼容的 TensorFlow：
pip install tensorflow-gpu

环境变量

确保您的环境变量配置正确：

echo 'export TF_CPP_MIN_LOG_LEVEL=2' >> ~/.bashrc
source ~/.bashrc

扩展阅读

如果您想了解更多关于 TensorFlow 在 Linux 环境下的配置和使用，可以参考以下链接：

图片

TensorFlow logo

TensorFlow Logo