本地安装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博客