Ubuntu下PyTorch安装步骤是什么

以下是在Ubuntu下安装PyTorch的步骤：

安装前准备

更新系统包列表：sudo apt update。若需GPU支持，安装NVIDIA显卡驱动，可通过sudo apt install nvidia-driver-xxx（xxx为驱动版本号）安装。同时，根据显卡型号从NVIDIA官网下载安装对应版本的CUDA工具包和cuDNN库。

使用Conda安装（推荐）

创建并激活虚拟环境：conda create -n pytorch_env python=3.8，conda activate pytorch_env。然后根据CUDA版本安装PyTorch，如conda install pytorch torchvision torchaudio cudatoolkit=12.0 -c pytorch（12.0替换为实际CUDA版本）。

使用pip安装

若不使用Conda，可先安装pip：sudo apt install python3-pip。再根据CUDA版本选择安装命令，如pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu120（12.0替换为实际CUDA版本），不使用GPU则用pip install torch torchvision torchaudio。

验证安装

在Python解释器中输入import torch，print(torch.__version__)查看版本号，print(torch.cuda.is_available())检查GPU是否可用。