解决:torch.cuda.is_available()一直返回False,显卡是NVIDA GeForce

目录

  • 1. 背景
  • 2. 发现问题根源
  • 3. 解决问题

1. 背景

AI时代了,之前一直不怎么用到的小米笔记本Pro的 NVIDA GeForce MX250独显,就想着让它发挥余热,免得买了这么多年,一直闲置,浪费。

无脑按照chatGPT给的例子,一段乱搞,突然发现一直报错,说无可用的GPU,这就烦人了,明明有,但是好像隐身了,why?

  • 打开cmd,输入命令:python,进入python的运行环境
  • 输入import torch,稍等一下等它加载
  • 再输入print(torch.cuda.is_available()),一直得到的结果是False

2. 发现问题根源

  • 打开cmd,输入命令:python,进入python的运行环境
  • 输入import torch,稍等一下等它加载
  • 再输入print(torch.__version__),输出torch的版本,可以看到是万恶的cpu版本,破案了!

3. 解决问题

  • 卸载torch,打开cmd,输入pip uninstall torch
  • 重新安装gpu版本的torch,具体可以参考《分享:使用 python 让 NVIDA GeForce MX250 显卡发挥余热》
  • 检查是否安装成功,打开cmd,输入命令:python,进入python的运行环境
  • 输入import torch,稍等一下等它加载
  • 再输入print(torch.cuda.is_available()),得到结果True,棒!

    enjoy!

本文链接:https://my.lmcjl.com/post/2057.html

展开阅读全文

4 评论

留下您的评论.