安装pytorch gpu

安装 PyTorch GPU 版本需要满足以下条件：

1. NVIDIA 显卡，支持 CUDA（Compute Unified Device Architecture）计算架构，可以使用 nvidia-smi 命令查看显卡信息。

2. 安装 CUDA 工具包和 cuDNN 库，这是使用 PyTorch GPU 版本的必要条件。

3. 安装 Anaconda 或者 Miniconda，可以方便地创建虚拟环境，隔离不同项目的依赖关系。

4. 创建一个虚拟环境并激活，例如：

   conda create -n pytorch-gpu python=3.7
   conda activate pytorch-gpu

接下来，可以使用以下命令安装 PyTorch GPU 版本：

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

其中，cudatoolkit=11.0 指定了 CUDA 工具包的版本，可以根据自己的实际情况进行调整。安装完成后，可以使用以下代码测试 PyTorch GPU 版本是否安装成功：

import torch
print(torch.cuda.is_available())

如果输出 True，则说明 PyTorch GPU 版本安装成功。