Ubuntu PyTorch安装步骤是什么

在Ubuntu上安装PyTorch的步骤如下：

安装准备工作

1. 更新系统软件包：

sudo apt update

2. 安装Python3-venv（可选，但推荐）：

sudo apt install python3-venv -y

3. 安装Anaconda（如果选择使用Anaconda安装PyTorch）：

• 下载Anaconda安装程序脚本：

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

• 验证下载文件的完整性：

sha256sum anaconda.sh

• 运行安装程序脚本：

bash anaconda.sh

• 激活Anaconda环境：

source ~/.bashrc

安装PyTorch

使用pip安装PyTorch（仅支持CPU）：

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

使用pip安装支持GPU的PyTorch（需要安装CUDA）：

1. 安装CUDA：

• 下载并安装CUDA Toolkit，具体版本请根据你的显卡和需求选择。可以从NVIDIA官网下载。

2. 安装cuDNN：

• 下载并安装与CUDA版本匹配的cuDNN库。可以从NVIDIA官网下载。

3. 创建并激活虚拟环境（可选，但推荐）：

conda create -n pytorch_env python=3.8
conda activate pytorch_env

4. 使用Conda安装PyTorch（支持CPU和GPU）：

• 仅支持CPU：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

• 支持GPU（安装CUDA与PyTorch）：

conda install pytorch torchvision torchaudio pytorch-cuda11.7 -c pytorch -c nvidia

验证安装

在Python解释器中运行以下命令来验证PyTorch是否安装成功：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 应该返回True，如果安装了GPU版本的PyTorch
print(torch.cuda.device_count())  # 应该返回可用的GPU数量

请注意，上述步骤可能会随着PyTorch版本的更新和系统环境的变化而有所不同。建议在安装前查阅PyTorch的官方文档以获取最新的安装指南。