Ubuntu如何配置PyTorch GPU加速

在Ubuntu上配置PyTorch GPU加速，可按以下步骤进行：

1. 安装NVIDIA驱动：添加NVIDIA PPA，更新软件包列表后安装适合GPU型号的驱动，如sudo add-apt-repository ppa:graphics-drivers/ppa，sudo apt update，sudo apt install nvidia-driver-<版本号>，安装后重启电脑。
2. 安装CUDA Toolkit：从NVIDIA官网下载对应Ubuntu版本的安装包，用dpkg命令安装，如sudo dpkg -i cuda-repo-<distro>_<version>_amd64.deb，安装后设置环境变量，在~/.bashrc中添加export PATH=/usr/local/cuda/bin:$PATH和export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH，并执行source ~/.bashrc。
3. 安装cuDNN库：注册NVIDIA开发者账户，下载与CUDA版本兼容的cuDNN库，解压后将文件复制到CUDA目录，如tar -xzvf cudnn-<version>-linux-x64-v*-*.tgz，sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/，sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/，并设置文件权限为可读。
4. 安装PyTorch：使用pip或conda安装与CUDA版本兼容的PyTorch，如pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu<版本号>或conda install pytorch torchvision torchaudio cudatoolkit=<版本号> -c pytorch。
5. 验证安装：在Python中运行import torch，再执行print(torch.cuda.is_available())，若输出为True，则配置成功。