pytorch下载 cuda12.1

要下载PyTorch并安装CUDA 12.1，可以按照以下步骤进行操作：

1. 首先，可以尝试添加清华源以提高网络下载速度。在命令行中输入以下命令：

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

2. 接下来，使用pip命令安装PyTorch。在命令行中输入以下命令：

pip install torch==2.0.0 cu121 torchvision==0.15.1 cu121 torchaudio==2.0.1 cu121 -f https://download.pytorch.org/whl/torch_stable.html

这将安装PyTorch 2.0.0版本以及与CUDA 12.1兼容的相关软件包。

3. 安装完成后，可以验证PyTorch是否成功安装并与CUDA 12.1一起使用。在Anaconda Prompt或命令行中输入以下命令：

python
>>> import torch
>>> torch.__version__
'2.0.0 cu121'
>>> torch.cuda.is_available()
True
>>> torch.cuda.get_device_name(0)
'NVIDIA GeForce RTX 3090'

如果出现上述输出，说明PyTorch已成功安装并与CUDA 12.1一起使用。