ubuntu安装 gpu torch

在Ubuntu上安装GPU支持的PyTorch，通常需要先确保你的系统已经安装了CUDA和cuDNN（用于加速深度学习运算），因为PyTorch依赖这两个库。以下是基本步骤：

1. **更新包列表并安装必要的软件：**

   sudo apt-get update
   sudo apt-get install -y software-properties-common

2. **添加CUDA和驱动源：**

   sudo add-apt-repository ppa:graphics-drivers/ppa
   sudo apt-get update

然后选择适合你的显卡型号的CUDA版本安装，例如：

   sudo apt-get install cuda cudnn

3. **安装PyTorch：**

   sudo pip install torch torchvision

如果你需要特定版本，可以指定版本号：

   sudo pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111

4. **检查安装：**

   python -c "import torch; print(torch.cuda.is_available())"

安装过程中可能会遇到权限问题，这时可能需要运行`sudo`。同时，记得根据你的显卡配置调整相应的命令。