温馨提示×

Ubuntu如何配置PyTorch进行深度学习研究

小樊
45
2025-09-09 01:33:30
栏目: 智能运维

以下是在Ubuntu上配置PyTorch进行深度学习研究的步骤:

系统准备

更新系统包列表:sudo apt update

安装Python和pip

安装Python和pip(若未安装):sudo apt install python3 python3-pip

创建虚拟环境(可选)

使用venvconda创建,如python3 -m venv pytorch -env,然后激活:source pytorch -env/bin/activate

安装PyTorch

  • CPU版本pip install torch torchvision torchaudio
  • GPU版本:先安装对应CUDA和cuDNN,再通过pip安装,如pip install torch torchvision torchaudio --extra -index -url https://download.pytorch.org/whl/cu117

验证安装

在Python解释器中运行:import torchprint(torch.__version__)print(torch.cuda.is_available())

安装其他依赖库(可选)

根据研究需求安装,如pip install numpy pandas matplotlib等。

0