PyTorch在Linux上的安装步骤

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

安装Anaconda

1. 访问Anaconda官网下载与操作系统匹配的版本安装包。
2. 安装时勾选“Add Anaconda to PATH”（Windows用户必须勾选，否则后续命令无法直接使用）。
3. 安装完成后，打开终端输入 conda --version，若显示版本号（如 conda 23.7.4），则安装成功。

创建虚拟环境

1. 创建虚拟环境命令：

conda create -n pytorch_env python=3.8

Python 3.8是兼容PyTorch的稳定版本，可替换为3.9或3.10（需确认PyTorch支持）。

2. 激活环境：

conda activate pytorch_env

激活后终端提示符应显示 (pytorch_env)。

安装PyTorch和CUDA（如果使用GPU）

CPU版本

conda install pytorch torchvision torchaudio cpuonly -c pytorch

GPU版本（需NVIDIA显卡和CUDA 11.8）

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

验证安装

1. 查看自己的CUDA型号，输入如下命令：

nvidia-smi

2. 从PyTorch官网找到合适的版本，复制相应的命令到终端进行安装。

验证PyTorch是否安装成功

在创建的小环境下输入 python，在Python解释器中输入以下代码，返回 True，安装成功：

import torch
torch.cuda.is_available()

以上步骤应该可以帮助你在Linux系统上成功安装PyTorch。如果在执行过程中遇到问题，可以参考PyTorch的官方文档或相关社区论坛寻求帮助。