cuda12.1安装pytorch

要在 CUDA 12.1 上安装 PyTorch，您可以按照以下步骤进行操作：

1. 确保您的系统已安装 CUDA 12.1。您可以从 NVIDIA 的官方网站下载并安装 CUDA Toolkit。

2. 创建一个虚拟环境（可选但推荐），以隔离您的 Python 环境。您可以使用 Anaconda 或 venv 等工具来创建虚拟环境。

3. 在虚拟环境中安装 PyTorch。您可以使用 pip 命令来安装适用于 CUDA 12.1 的 PyTorch 版本。请注意，PyTorch 提供了不同的安装选项，具体取决于您的系统配置和需求。以下是一个示例安装命令：

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

请根据您的需求和系统配置进行调整。

4. 安装完成后，您可以在 Python 脚本或交互式环境中导入 PyTorch 并开始使用它：

import torch

# 测试 PyTorch 是否成功安装
print(torch.__version__)

这样，您就成功在 CUDA 12.1 上安装了 PyTorch。请确保您的系统满足 PyTorch 的最低要求，并参考官方文档获取更多细节和其他配置选项。