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

在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())

进行深度学习项目

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