ubuntu

PyTorch在Ubuntu上的依赖怎么安装

小樊
38
2025-08-24 00:38:29
栏目: 智能运维

在Ubuntu上安装PyTorch依赖,可参考以下步骤:

  1. 更新系统:在终端运行sudo apt update && sudo apt upgrade -y,确保软件包列表最新。
  2. 安装Python和pip:执行sudo apt install python3 python3-dev python3-pip -y,安装Python 3及其开发工具和pip。
  3. 创建虚拟环境(可选):使用python3 -m venv myenv创建名为myenv的虚拟环境,然后通过source myenv/bin/activate激活它,以隔离项目依赖。
  4. 安装PyTorch:根据是否使用GPU,选择不同安装命令。CPU版本:pip install torch torchvision torchaudio;GPU版本需先安装CUDA和cuDNN,再执行pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117(以CUDA 11.7为例)。

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

0
看了该问题的人还看了