【pytorch】torch.cuda.is_available() = False 可能的解决办法

原创已于 2022-11-05 13:57:18 修改 · 288 阅读

CC 4.0 BY-SA版权

文章标签：

于 2022-11-05 13:52:47 首次发布

42 篇文章

订阅专栏

本文介绍如何解决因网络问题导致的PyTorch CPU版本误装问题，并提供手动下载及安装GPU版本的具体步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

cuda 版本：11.6
cudnn 版本：11.5

问题原因

之前安装 torch 的时候由于网络原因找不到官方的 whl，导致 Pip 自动安装 cpu 版本的 torch

手动去官网下载下载链接中的 whl ，然后手动安装 whl

torch-1.12.0+cu116-cp37-cp37m-win_amd64
torchaudio-0.12.0+cu116-cp37-cp37m-win_amd64
torchvision-0.13.0+cu116-cp37-cp37m-win_amd64.whl