Debian下PyTorch的GPU支持情况如何

Debian下PyTorch支持GPU加速，需安装NVIDIA显卡驱动、CUDA Toolkit和cuDNN库，具体如下：

1. 驱动安装：通过apt安装NVIDIA驱动，如sudo apt install nvidia-driver-<version>，需根据显卡型号选择版本。
2. CUDA安装：从NVIDIA官网下载对应Debian版本的安装包（如.deb文件），通过dpkg安装，需配置仓库密钥。
3. cuDNN安装：注册NVIDIA开发者账号后下载，解压并将文件复制到CUDA目录（/usr/local/cuda）。
4. PyTorch安装：通过pip指定CUDA版本安装，如pip install torch --extra-index-url https://download.pytorch.org/whl/cu117（对应CUDA 11.7）。
5. 验证：运行import torch; print(torch.cuda.is_available())，返回True表示GPU支持正常。

注：需确保CUDA、cuDNN与PyTorch版本兼容，可参考PyTorch官网的版本对照表。