PyTorch能直接在Ubuntu用吗

PyTorch可以在Ubuntu上直接使用，需先安装Python和pip，再通过pip安装PyTorch及其相关库（如torchvision、torchaudio）。根据是否需要GPU加速，可选择安装CPU或CUDA版本：

• CPU版本：直接使用pip install torch torchvision torchaudio命令安装。
• GPU版本：需先安装NVIDIA驱动、CUDA和cuDNN，再通过指定CUDA版本的pip命令安装，例如pip install torch==2.6.0+cu126 torchvision==0.17.0+cu126 torchaudio==2.6.0+cu126 --index-url https://download.pytorch.org/whl/cu126。

安装完成后，可通过import torch和print(torch.__version__)验证是否安装成功。