正文

Ubuntu 18.04轻松安装TensorFlow框架全攻略

/0 浏览量
0716

在本文中，我们将一步步教你如何在Ubuntu 18.04系统上轻松安装TensorFlow框架。TensorFlow是一个开源的机器学习框架，由Google开发，用于广泛的应用，包括数据流编程和不同的神经网络结构。下面，让我们开始吧！

准备工作

在开始之前，请确保你的Ubuntu 18.04系统已经更新了软件包列表：

sudo apt update
sudo apt upgrade

安装依赖项

TensorFlow需要一些依赖项，包括Python和pip。首先，我们需要安装Python和pip：

sudo apt install python3 python3-pip

接着，升级pip到最新版本：

sudo pip3 install --upgrade pip

安装TensorFlow

TensorFlow支持多种Python版本和操作系统。在Ubuntu 18.04上，你可以使用以下命令安装TensorFlow：

CPU版本

如果你没有GPU，你可以安装CPU版本的TensorFlow：

pip3 install tensorflow

GPU版本

如果你有NVIDIA GPU，你可以安装支持GPU加速的TensorFlow版本：

pip3 install tensorflow-gpu

安装过程中可能会遇到一些问题，例如缺少CUDA库或cuDNN库。下面我们将详细介绍如何解决这些问题。

解决依赖项问题

安装CUDA

首先，你需要安装CUDA。CUDA是一个用于并行计算的开源平台，由NVIDIA提供。以下是安装CUDA的步骤：

  1. 访问NVIDIA CUDA Toolkit官网：https://developer.nvidia.com/cuda-downloads
  2. 下载适合你NVIDIA GPU的CUDA Toolkit版本。
  3. 解压下载的文件。
  4. 打开终端，切换到解压后的文件夹。
  5. 运行以下命令安装CUDA：
sudo ./cuda_11.x.x_linux.run

注意替换11.x.x为实际版本号。

安装cuDNN

cuDNN是NVIDIA提供的深度学习库，用于加速TensorFlow等深度学习框架。以下是安装cuDNN的步骤：

  1. 访问NVIDIA cuDNN官网：https://developer.nvidia.com/cudnn
  2. 根据你的CUDA版本选择相应的cuDNN版本。
  3. 下载cuDNN文件。
  4. 解压下载的文件。
  5. 将cuDNN库复制到相应的文件夹：
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
  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

测试TensorFlow安装

安装完成后，我们可以通过运行以下Python代码来测试TensorFlow是否正常工作：

import tensorflow as tf

print(tf.__version__)

如果输出TensorFlow的版本号，说明TensorFlow已经成功安装。

总结

现在，你已经成功在Ubuntu 18.04系统上安装了TensorFlow框架。你可以开始使用TensorFlow进行机器学习项目了。希望本文对你有所帮助！

-- 展开阅读全文 --