温馨提示×

如何在Debian上使用PyTorch进行深度学习

小樊
57
2025-08-06 05:04:14
栏目: 智能运维

在Debian上使用PyTorch进行深度学习,可按以下步骤操作:

安装依赖

更新系统包列表,安装Python、pip及编译工具等依赖:

sudo apt update
sudo apt install -y python3 python3-pip build-essential

安装PyTorch

可根据是否使用GPU选择安装方式:

  • CPU版本:直接用pip安装,命令为pip install torch torchvision torchaudio
  • GPU版本:若有NVIDIA GPU,需先安装对应版本的CUDA和cuDNN,再通过pip安装支持CUDA的PyTorch,如pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117(以CUDA 11.7为例)。也可使用conda安装,命令为conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch

验证安装

在Python环境中运行以下代码,查看PyTorch版本及是否支持CUDA:

import torch
print(torch.__version__)
print(torch.cuda.is_available())

进行深度学习项目

安装好后,可按照深度学习项目的基本流程进行操作,如数据集加载与预处理、定义模型、训练和测试模型等。

0