如安在 Ubuntu 上安装 PyTorch

简介

PyTorch 因其易用性、动态盘算图和高效性而日益盛行，成为实现深度学习模型的首选。如果你想探索这个工具并学习如安在 Ubuntu 上安装 PyTorch，本指南将对你有所资助！
在本教程中，我们将引导你完成在 Ubuntu 体系上使用 Pip 或 Anaconda 安装 PyTorch 的过程，资助你开始利用这个工具。
安装和配置步骤

第一步：更新体系软件包

首先，更新体系软件包，以确保你使用的是最新的软件包。

1. sudo apt update

复制代码

第二步：安装 Python3-venv

接下来，你须要安装 Python3-venv，你可以使用它为你的项目创建一个隔离的 Python 情况，这样你就不必全局安装 Python 包，从而防止大概的兼容性问题。

1. sudo apt install python3-venv -y

复制代码

第三步：设置情况

现在创建并导航到一个新的目录，该目录将用于存储捏造情况文件和设置。这里 “pytorch_env” 是目录的名称；如果须要，你可以使用不同的名称。

1. mkdir pytorch_env
2. cd pytorch_env

复制代码

接下来，运行以下命令来创建并激活捏造情况。

1. python3 -m venv pytorch_env
2. source pytorch_env/bin/activate

复制代码

一旦捏造情况被激活，你应该看到你的命令提示符发生变化，显示捏造情况的名称。激活此情况后，你可以安装与此捏造情况干系的 Python 包或脚本，而不是体系范围的 Python，从而避免与须要不同包版本的项目发生冲突。

第四步：使用 Pip 安装 PyTorch

现在，你的捏造情况已激活，你可以继续在 Ubuntu 体系上安装 PyTorch。
仅支持 CPU：
要仅安装支持 CPU 的 PyTorch，请运行：

1. pip3 install torch torchvision torchaudio
2. --index-url <https://download.pytorch.org/whl/cpu>

复制代码

支持 GPU (安装 CUDA 与 PyTorch) (CUDA)：
要安装支持 GPU 的 PyTorch，请运行：

1. pip3 install torch torchvision torchaudio

复制代码

第五步：验证安装

你可以运行以下命令来验证 PyTorch 是否已安装：

1. python
2. import torch
3. print(torch.__version__)

复制代码

上面的命令将启动 Python 表明器，导入 PyTorch 库并打印当前安装的 PyTorch 版本。

使用 Conda 安装 PyTorch

你还可以使用名为 Anaconda 的开源平台来安装 PyTorch。 你可以通过以下步骤使用 Conda 安装 PyTorch：
第一步：更新体系软件包

像往常一样，你须要确保你的体系软件包是最新的。

1. sudo apt update

复制代码

第二步：安装 Anaconda

接下来，你须要安装 Anaconda，你可以按照以下步骤进行安装。
首先，你须要安装 Curl。这将用于下载 Anaconda 安装程序脚本。

1. sudo apt install curl -y

复制代码

接下来，作为最佳实践，你须要隔离安装过程。导航到 “tmp” 目录，如果你须要，你可以创建或使用不同的目录。

1. cd /tmp

复制代码

接下来，通过运行以下命令，使用 Curl 下载 Anaconda 安装程序脚本：

1. curl --output anaconda.sh https://repo.anaconda.com/archive/Anaconda3-2023.07-2-Linux-x86_64.sh

复制代码

上面的命令将下载 Anaconda 安装程序脚本并将其保存为名为 “anaconda.sh” 的文件。你始终可以在他们的网站上获取最新版本脚本的 URL：https://repo.anaconda.com/archive。

现在使用以下命令验证下载文件的完整性：

1. sha256sum anaconda.sh

复制代码

你须要确认 sha256sum 校验和值与官方网站上的值匹配：https://repo.anaconda.com/archive。

验证下载脚本后，你可以继续运行 Anaconda 安装程序脚本并通过运行以下命令开始安装过程：

1. bash anaconda.sh

复制代码

安装过程完成后，你须要更新当前 shell 会话，以确保 Anaconda 情况及其命令可用于当前的终端会话。通过运行以下命令执行此操纵：

1. source ~/.bashrc

复制代码

它还会激活 Anaconda，正如你在提示符更改为 “base” 时所看到的那样，这是 Anaconda 在安装过程中创建的默认情况。 随后，你可以使用 conda activate 命令来激活 Anaconda。

你可以使用 conda --version 命令来验证 Anaconda 是否已安装。

第三步：使用 Anaconda 安装 PyTorch

现在，Anaconda 已安装并激活，你可以使用它来安装 PyTorch。
仅支持 CPU：

1. conda install pytorch torchvision torchaudio cpuonly -c pytorch

复制代码

支持 GPU (安装带有 CUDA 的 PyTorch)：

1. conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

复制代码

第四步：验证安装

运行以下命令来验证 PyTorch 是否已安装：

1. python
2. import torch
3. print(torch.__version__)

复制代码

怎样卸载 PyTorch？

如果你须要在 Ubuntu 体系上卸载 PyTorch，你可以通过运行以下命令使用 Pip 来执行此操纵：

1. pip3 uninstall torch -y

复制代码

要使用 Anaconda 卸载 PyTorch，请运行：

1. conda remove pytorch -y

复制代码

末端

至此，你的体系上应该已经安装了 PyTorch。本指南展示了怎样使用 Pip 和 Anaconda 在 Ubuntu 上安装 PyTorch。 你还相识到 PyTorch 提供了基于体系上可用盘算资源下载的选项。 在办理了 PyTorch 的安装问题后，你可以继续探索其功能并体验其效率。
我的博客：https://blog.ivwv.site

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。