温馨提示×

PyTorch能直接在Ubuntu用吗

小樊
62
2025-08-22 00:22:18
栏目: 智能运维

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 torchprint(torch.__version__)验证是否安装成功。

0