尚未崩坏 发表于 2025-3-16 18:06:14

如安在 Ubuntu 上安装 PyTorch

简介

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

第一步:更新体系软件包

首先,更新体系软件包,以确保你使用的是最新的软件包。
sudo apt update

第二步:安装 Python3-venv

接下来,你须要安装 Python3-venv,你可以使用它为你的项目创建一个隔离的 Python 情况,这样你就不必全局安装 Python 包,从而防止大概的兼容性问题。
sudo apt install python3-venv -y
https://i-blog.csdnimg.cn/img_convert/ac962076aabaa376a1df2dafe8cf7ca7.png
第三步:设置情况

现在创建并导航到一个新的目录,该目录将用于存储捏造情况文件和设置。这里 “pytorch_env” 是目录的名称;如果须要,你可以使用不同的名称。
mkdir pytorch_env
cd pytorch_env
https://i-blog.csdnimg.cn/img_convert/b5be589994526191d34a294518c26df4.png
接下来,运行以下命令来创建并激活捏造情况。
python3 -m venv pytorch_env
source pytorch_env/bin/activate
一旦捏造情况被激活,你应该看到你的命令提示符发生变化,显示捏造情况的名称。激活此情况后,你可以安装与此捏造情况干系的 Python 包或脚本,而不是体系范围的 Python,从而避免与须要不同包版本的项目发生冲突。
https://i-blog.csdnimg.cn/img_convert/71a09d15aa9e15568ef8b34a8ac83bad.png
第四步:使用 Pip 安装 PyTorch

现在,你的捏造情况已激活,你可以继续在 Ubuntu 体系上安装 PyTorch。
仅支持 CPU:
要仅安装支持 CPU 的 PyTorch,请运行:
pip3 install torch torchvision torchaudio
--index-url <https://download.pytorch.org/whl/cpu>
https://i-blog.csdnimg.cn/img_convert/cabab3385e049a9c0016961f864e6205.png
支持 GPU (安装 CUDA 与 PyTorch) (CUDA):
要安装支持 GPU 的 PyTorch,请运行:
pip3 install torch torchvision torchaudio
第五步:验证安装

你可以运行以下命令来验证 PyTorch 是否已安装:
python
import torch
print(torch.__version__)
上面的命令将启动 Python 表明器,导入 PyTorch 库并打印当前安装的 PyTorch 版本。
https://i-blog.csdnimg.cn/img_convert/870245bd3ca315318b6e80633b585325.png
使用 Conda 安装 PyTorch

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

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

第二步:安装 Anaconda

接下来,你须要安装 Anaconda,你可以按照以下步骤进行安装。
首先,你须要安装 Curl。这将用于下载 Anaconda 安装程序脚本。
sudo apt install curl -y
https://i-blog.csdnimg.cn/img_convert/cd8bbe0efa444b6b45aad0a0e86c2cfe.png
接下来,作为最佳实践,你须要隔离安装过程。导航到 “tmp” 目录,如果你须要,你可以创建或使用不同的目录。
cd /tmp
https://i-blog.csdnimg.cn/img_convert/a299736ac7fdf9625884637f489b3ea1.png
接下来,通过运行以下命令,使用 Curl 下载 Anaconda 安装程序脚本:
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。
https://i-blog.csdnimg.cn/img_convert/ad634acdb5482294f86243d9419db0ee.png
现在使用以下命令验证下载文件的完整性:
sha256sum anaconda.sh
你须要确认 sha256sum 校验和值与官方网站上的值匹配:https://repo.anaconda.com/archive。
https://i-blog.csdnimg.cn/img_convert/8ed309b53e4eeb2ae5448b1d29d9be53.png
验证下载脚本后,你可以继续运行 Anaconda 安装程序脚本并通过运行以下命令开始安装过程:
bash anaconda.sh
https://i-blog.csdnimg.cn/img_convert/ee12f1c011686b5befe0c4ec73de0600.png
安装过程完成后,你须要更新当前 shell 会话,以确保 Anaconda 情况及其命令可用于当前的终端会话。通过运行以下命令执行此操纵:
source ~/.bashrc
它还会激活 Anaconda,正如你在提示符更改为 “base” 时所看到的那样,这是 Anaconda 在安装过程中创建的默认情况。 随后,你可以使用 conda activate 命令来激活 Anaconda。
https://i-blog.csdnimg.cn/img_convert/7c1225f35cbef75c64620be885529058.png
你可以使用 conda --version 命令来验证 Anaconda 是否已安装。
https://i-blog.csdnimg.cn/img_convert/99dcfc16ed1180fb895e390aef812c99.png
第三步:使用 Anaconda 安装 PyTorch

现在,Anaconda 已安装并激活,你可以使用它来安装 PyTorch。
仅支持 CPU:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
https://i-blog.csdnimg.cn/img_convert/3bfcf62b43b61d9f7870f7ad6bde16ec.png
支持 GPU (安装带有 CUDA 的 PyTorch):
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
第四步:验证安装

运行以下命令来验证 PyTorch 是否已安装:
python
import torch
print(torch.__version__)
https://i-blog.csdnimg.cn/img_convert/f5f9c89c44d3820bfe96d81e24988778.png
怎样卸载 PyTorch?

如果你须要在 Ubuntu 体系上卸载 PyTorch,你可以通过运行以下命令使用 Pip 来执行此操纵:
pip3 uninstall torch -y
https://i-blog.csdnimg.cn/img_convert/b8d98d80fd4685266dbe8bcc51d7f454.png
要使用 Anaconda 卸载 PyTorch,请运行:
conda remove pytorch -y
https://i-blog.csdnimg.cn/img_convert/d53643a8313a66513828e6637a265cd5.png
末端

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

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: 如安在 Ubuntu 上安装 PyTorch