Ubuntu安装pytorch GPU版本

安装PyTorch GPU版本需要先安装CUDA和cuDNN，然后使用pip安装PyTorch。这里是安装步骤：

1. 安装CUDA：首先需要去NVIDIA的官网下载对应版本的CUDA并安装。可以在终端输入`nvcc -V`来检查CUDA是否安装成功。

2. 安装cuDNN：然后需要去NVIDIA的官网下载对应版本的cuDNN并安装。可以在终端输入`cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2`来检查cuDNN是否安装成功。

3. 安装PyTorch：使用pip安装PyTorch GPU版本，可以使用以下命令：

   pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

其中`-f`选项指定了使用的PyTorch版本，这里使用的是支持CUDA 11.1的版本，可以根据自己的CUDA版本选择对应的版本。

4. 测试：可以在Python中导入PyTorch并测试GPU是否可用：

   import torch
   
   if torch.cuda.is_available():
       device = torch.device("cuda")
       print(f'There are {torch.cuda.device_count()} GPU(s) available.')
       print(f'GPU {torch.cuda.current_device()} is selected.')
   else:
       device = torch.device("cpu")
       print('No GPU available, using CPU instead.')
   
   # 输出
   # There are 1 GPU(s) available.
   # GPU 0 is selected.

如果输出结果中显示有可用的GPU，说明安装成功。