当前位置: 首页 > news >正文

PyTorch和cude版本不兼容导致无法检测到GPU

具体问题如下:
环境:windows 11 + conda
pytorch 已经安装,但 cude 不可用,可用gpu数量为0
使用whisper时提示错误信息:

... ...
AssertionError: Torch not compiled with CUDA enabled

原因:
gpu驱动程序支持pytorch版本有上限,而并不是越新越好。
运行nvidia-smi结果右上角的CUDA Version就是版本上限。
要安装小于此版本号的torch而不是最新版。
具体方法参考下文python注释。

# pytorch_check.pyimport torchif torch.cuda.is_available():print(f"{torch.cuda.device_count()}个GPU 可用:", torch.cuda.get_device_name(0))
else:print("未检测到N卡GPU")print("""## 检查NVIDIA驱动状态
运行 nvidia-smi 确认驱动是否正常安装且支持CUDA。若命令无输出,需重新安装或更新驱动。
驱动版本需与PyTorch要求的CUDA版本兼容(如驱动支持的最高CUDA版本需≥PyTorch所需版本)。## 验证CUDA工具包安装
通过 nvcc --version 检查CUDA编译器版本是否与PyTorch匹配。
若未安装CUDA工具包,需从 [Nvidia官网](https://developer.nvidia.com/cuda-toolkit-archive) 下载对应版本。## 具体方法
运行 `nvidia-smi` 查看你的显卡驱动支持的**最高**CUDA版本, 不一定是实际安装的CUDA Toolkit版本
如果显示12.6,就要安装小于等于12.6版本的torch (目前PyTorch稳定版支持的CUDA版本主要为 11.8 和 12.1,而CUDA 12.6可能尚未被官方支持)`conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia`或使用pip安装:`pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121`可能需要先清理之前的版本:`conda uninstall pytorch torchvision torchaudio`""")
http://www.hskmm.com/?act=detail&tid=14637

相关文章:

  • 嵌入式系统arm高级系统调试技能-24./proc/slabinfo 记录解读与内存异常分析
  • 2025无人机在低空物流中的应用实践
  • ssh如何打开可视化界面
  • NanoCAD 24.0安装包下载地址与安装教程
  • 改 187 个接口参数:Postman 卡壳时,Apipost 凭什么 5 分钟搞定?
  • 使用AWS Amplify、Lambda、API Gateway和DynamoDB部署静态Web应用
  • vscode的ssh-remote插件经常掉线
  • 记录第一次CCPC(2025)网络赛前后
  • 第四周课前思考
  • 声像新境:东芝电视以火箭炮SOUND重塑家庭艺术馆新标准
  • c语言数组与指针
  • 开发微信机器人/微信协议/个人微信api接口
  • 深入解析:frp实现内网穿透,公网服务器或云服务器配置frps,本地内网配置frpc
  • 【五行】根据天干、地支、生肖起姓名(9月出生的宝宝可参考)
  • 全差分放大器(FDA)电路设计计算问题及电压范围估算[原创www.cnblogs.com/helesheng]
  • 使用WTAPI开发智能微信机器人文档
  • [Android]自定义view - 详解
  • 不定高元素动画实现方案(下)
  • 详细介绍:C 语言:第 20 天笔记:typedef(类型重命名规则、应用场景与实战案例)
  • Screaming Architecture:让架构自己说话
  • BOE(京东方)携手UNESCO联合主办WCBR“科学十年”分会 彰显中国科技企业可持续发展实力
  • 使用Cyclops.PdfKit根据pdf模板生成pdf文件
  • 一款文本编辑器的介绍
  • 随笔-决战保研篇
  • 科研人必知:293F与HEK293细胞在蛋白表达中的不同“超能力”
  • Redis Cluster
  • 如何使用C语言实现Vigenre密码加解密
  • 【F#学习】列表 List
  • Trae与Gitee MCP深度集成:AI编程工具链迎来重大升级
  • 【2025-09-22】加班感悟