python3.9安装pytorch gpu

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

1. 您的电脑必须有 NVIDIA 显卡，且显卡型号不低于 Kepler 架构（即 NVIDIA GeForce GTX 6XX 或更高版本）。
2. 您已经安装了 CUDA 和 cuDNN。

接下来，您可以按照以下步骤安装 PyTorch GPU 版本：

1. 打开终端或 Anaconda Prompt，输入以下命令创建一个新的 conda 环境并激活它：

conda create --name pytorch-gpu python=3.9
conda activate pytorch-gpu

2. 在 conda 环境中安装 PyTorch GPU 版本，您可以根据自己的 CUDA 版本和显卡型号选择合适的版本，例如：

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

其中，`cudatoolkit=11.1` 表示使用 CUDA 11.1 版本，您可以根据自己的 CUDA 版本进行修改。

3. 安装完成后，您可以在 Python 中导入 PyTorch 并测试是否安装成功：

import torch

if torch.cuda.is_available():
    device = torch.device("cuda")
    print("GPU is available")
else:
    device = torch.device("cpu")
    print("GPU is not available")

如果输出结果为 `GPU is available`，则说明 PyTorch GPU 版本安装成功。