Ubuntu上PyTorch的GPU支持情况

Ubuntu上PyTorch支持GPU。需安装NVIDIA显卡驱动、CUDA Toolkit、cuDNN库，再安装对应版本的PyTorch GPU版本，具体步骤如下：

1. 安装NVIDIA驱动：通过命令sudo apt update和sudo apt install nvidia-driver-版本号安装，版本号根据显卡型号确定。
2. 安装CUDA Toolkit：从NVIDIA官网下载与驱动和PyTorch兼容的版本，按官方指南安装。
3. 安装cuDNN库：下载与CUDA版本匹配的cuDNN，解压后复制文件到CUDA安装目录。
4. 安装PyTorch：用pip或conda安装，指定CUDA版本，如pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118。

安装后，可通过代码import torch; print(torch.cuda.is_available())验证，若输出True，则表示配置成功。