正文

教你轻松安装Ubuntu16.04深度学习框架：从TensorFlow到PyTorch，一步到位教程

/0 浏览量
0718

在探索深度学习领域时，选择合适的框架至关重要。Ubuntu 16.04作为一个稳定的操作系统，非常适合深度学习环境搭建。本文将带你轻松安装Ubuntu 16.04上的深度学习框架，包括TensorFlow和PyTorch，让你一步到位，开始深度学习之旅。

准备工作

在开始安装之前，请确保你的Ubuntu 16.04系统已经更新到最新状态：

sudo apt update
sudo apt upgrade

此外，为了后续操作方便，建议安装一些常用的工具：

sudo apt install build-essential python3-dev python3-pip python3-setuptools

安装TensorFlow

TensorFlow是Google开发的开源机器学习框架，广泛应用于深度学习领域。以下是安装TensorFlow的步骤：

  1. 创建虚拟环境：
python3 -m venv tensorflow_env
source tensorflow_env/bin/activate
  1. 安装TensorFlow：
pip install tensorflow==1.15

注意：这里选择TensorFlow 1.15版本，因为它是最后一个支持CPU版本的稳定版本。

  1. 验证安装：
import tensorflow as tf
print(tf.__version__)

如果输出版本信息，则说明TensorFlow安装成功。

安装PyTorch

PyTorch是Facebook开发的开源深度学习框架，以其易用性和灵活性受到广泛欢迎。以下是安装PyTorch的步骤：

  1. 安装PyTorch：
pip install torch torchvision torchaudio
  1. 验证安装：
import torch
print(torch.__version__)

如果输出版本信息，则说明PyTorch安装成功。

配置CUDA

为了在GPU上运行TensorFlow和PyTorch，需要安装CUDA。以下是安装CUDA的步骤：

  1. 下载CUDA Toolkit：

前往NVIDIA CUDA Toolkit官网下载适合你的CUDA Toolkit版本。

  1. 安装CUDA Toolkit：
sudo sh cuda_11.0.1_410.48_linux.run

根据提示完成安装。

  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. 验证CUDA安装：
nvcc --version

如果输出CUDA版本信息，则说明CUDA安装成功。

总结

通过本文的教程，你已经成功在Ubuntu 16.04上安装了TensorFlow和PyTorch，并配置了CUDA。现在你可以开始使用这些框架进行深度学习了。祝你在深度学习领域取得丰硕的成果！

-- 展开阅读全文 --