卸载之前安装的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博客

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐