成功解决:torch.cuda.is_available()返回False

from 老和山下小菜狗

问题:

import torch是没有问题的,但是在运行torch.cuda.is_available()之后返回值却是False。

解决方法:

  1. 这里下载对应cuda版本的torch和torchvision,如图所示。
    在这里插入图片描述
  2. 打开cmd终端,输入
pip install 文件名1
pip install 文件名2

如图所示。

在这里插入图片描述

  1. 在重装torch 和torchvision之后,torch.cuda.is_available()的返回值变为True。

猜你喜欢

转载自blog.csdn.net/shine_00/article/details/121038768