本地安装torch2.3.1,cuda12.1,python3.10
卸载之前安装的torch (torch2.2.1+cpu)
pip uninstall torch
pip uninstall torch下载对应的torch版本及相关文件
torch-2.3.1+cu121-cp310
torchvision-0.18.1+cu121
torchaudio-2.3.1+cu121
下载地址：pytorch-wheels-cu121安装包下载_开源镜像站-阿里云 (aliyun.com)
或百度网盘下载：
链接: https://pan.baidu.com/s/1n0ijikRUSaCCyk310J-b9g?pwd=pa3m 提取码: pa3m
分别安装
pip install "torch-2.3.1+cu121-cp310-cp310-win_amd64.whl"
pip install "torchaudio-2.3.1+cu121-cp310-cp310-win_amd64.whl"
pip install "torchvision-0.18.1+cu121-cp310-cp310-win_amd64.whl"检查是否安装成功，进入python后输入
import torch
print(torch.__version__)
print(torch.cuda.is_available())第一个打印 2.3.1+cu121，第二个打印True
参考
使用阿里云镜像本地安装torch的cu121版本示例_torch 2.3.1+cu121-CSDN博客