pytorch安装 gpu

要安装PyTorch的GPU版本，首先需要确保你已经安装了NVIDIA的显卡驱动程序和CUDA工具包。然后，你可以通过以下步骤来安装PyTorch的GPU版本：

1. 打开Anaconda Prompt或者终端，并进入到你的虚拟环境中（如果你使用了虚拟环境的话）。

2. 运行以下命令来安装PyTorch和torchvision的GPU版本：

   conda install pytorch torchvision cudatoolkit=<your_cuda_version>

注意将`<your_cuda_version>`替换为你安装的CUDA工具包的版本号，例如`10.1`。

3. 等待安装过程完成，安装完成后，你就成功地安装了PyTorch的GPU版本。

为了测试PyTorch和CUDA是否安装成功，你可以在PyCharm或者其他Python开发环境中打开一个新的Python文件，并输入以下代码：

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

运行代码后，如果输出结果为`True`，则表示PyTorch和CUDA安装成功，你的电脑支持GPU加速。