当前位置: 首页 > 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.rkmt.cn/news/10233.html

相关文章:

  • 2025无人机在低空物流中的应用实践
  • ssh如何打开可视化界面
  • 深入解析:MES系统在不同制造行业中的应用差异与共性
  • 使用AWS Amplify、Lambda、API Gateway和DynamoDB部署静态Web应用
  • 详细介绍:广告点击场景下的 YOLO-world 模型训练失败及其解决办法
  • 【深度学习计算机视觉】06:目标检测内容集
  • debian13 无法正常从休眠中恢复
  • 全差分放大器(FDA)电路设计计算问题及电压范围估算[原创www.cnblogs.com/helesheng]
  • 详细介绍:Java与Vue构建资产设备全周期管理系统,覆盖采购、入库、使用、维护至报废全流程,支持移动端实时操作与后台智能管理,提供完整源码便于二次开发
  • 使用WTAPI开发智能微信机器人文档
  • Zabbix7 监控USG6300E 并发IPv4会话数 - 教程
  • 详细介绍:C 语言:第 20 天笔记:typedef(类型重命名规则、应用场景与实战案例)
  • 企业如何选择适合自身行业的ERP系统?
  • Screaming Architecture:让架构自己说话
  • BOE(京东方)携手UNESCO联合主办WCBR“科学十年”分会 彰显中国科技企业可持续发展实力
  • 二级制流量算法熵值计算,N-Gram 算法(二:改进) - 教程
  • 随笔-决战保研篇
  • 【F#学习】列表 List
  • 嵌入式ARM架构学习9——IIC - 教程
  • 深入解析:GPT-5风波后,奥特曼深夜道歉并补偿!Plus用户成最大赢家,Wildcard 野卡停服后,普通人如何充值?
  • 完整教程:飞算JavaAI 2.0.0测评:自然语言编程如何颠覆传统开发?
  • .Net-IIS 文件上传安全漏洞问题
  • 【光照】[高光反射specular]以UnityURP为例
  • 游戏性能优化与逆向分析技术
  • 使用 feign 调用时对微服务实例进行选择
  • EI目录今年第3次更新!55本中国期刊被收录,附完整版下载
  • 自定义属性编辑器
  • Computer Graphics Tutorial
  • CF1874(CF Round 901) 总结
  • 2. Spring AI 快速入门使用 - Rainbow