PyTorch在Debian上的GPU支持情况如何

PyTorch在Debian上支持GPU加速，需手动配置CUDA和cuDNN，具体如下：

1. 系统要求：需安装NVIDIA显卡驱动（版本≥515.65.01，对应CUDA 11.7+）。
2. 安装CUDA Toolkit：从NVIDIA官网下载对应Debian版本的安装包，通过dpkg安装。
3. 安装cuDNN：下载与CUDA版本兼容的cuDNN库，复制文件至CUDA目录并设置权限。
4. 安装PyTorch GPU版：通过pip指定CUDA版本安装，如pip install torch --index-url https://download.pytorch.org/whl/cu118（对应CUDA 11.8）。
5. 验证：运行import torch; print(torch.cuda.is_available())，输出True表示配置成功。

注：Debian 11及以上版本支持CUDA，需确保驱动、CUDA、cuDNN版本匹配，建议参考PyTorch官网生成安装命令。