怎么下载pytorch（cuda11.7对应的）

你可以通过以下步骤下载并安装与 CUDA 11.7 对应的 PyTorch：

1. 首先，确保你已经安装了 CUDA 11.7 和对应的 cuDNN。

2. 接下来，打开 PyTorch 官网的安装页面：https://pytorch.org/get-started/locally/

3. 在页面上方的“Current stable release”下拉菜单中选择适合你操作系统的版本，然后选择“Stable”作为软件版本，最后选择“CUDA 11.1”作为 CUDA 版本。这是因为 PyTorch 官方并没有提供与 CUDA 11.7 直接对应的版本，但是 CUDA 11.1 和 CUDA 11.7 之间是兼容的。

4. 根据你的操作系统和安装方式选择适合你的命令。比如，如果你使用的是 Linux，pip 安装方式，并且想要安装 CPU 版本，可以使用以下命令：

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

如果你想要安装 GPU 版本，可以使用以下命令：

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

5. 最后，等待安装完成。安装完成后，你可以在 Python 中导入 PyTorch 并开始使用了。