pytorch应该安装哪个nvcc -V 还是 nvidia-smi 对比的cuda?

当使用nvidia-smi时会显示 cuda driver版本,如下:

NVIDIA-SMI 510.47.03 Driver Version: 510.47.03 CUDA Version: 11.6

当使用 nvcc -V 时会显示 cuda runtime version,如下:

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2020 NVIDIA Corporation

Built on Mon_Nov_30_19:08:53_PST_2020

Cuda compilation tools, release 11.2, V11.2.67

Build cuda_11.2.r11.2/compiler.29373293_0

这时候两个cuda版本时不一样,该使用哪个?

应该使用 nvcc -V 这个对应的cuda版本,这样就会兼容。

参考:GET was unable to find an engine to execute this computation · Issue #4060 · ultralytics/ultralytics · GitHub

相关推荐
浊酒南街1 分钟前
吴恩达深度学习笔记:卷积神经网络(Foundations of Convolutional Neural Networks)2.7-2.8
人工智能·深度学习·神经网络
qq_273900235 分钟前
解析TMalign文本文件中的转换矩阵
python·生物信息学
DuoRuaiMiFa36 分钟前
ChatGPT全新功能Canvas上线:开启智能编程与写作新篇章
人工智能·chatgpt
DisonTangor39 分钟前
Windows 11将新增基于AI的搜索、生成式填充和其它AI功能
人工智能
soso196841 分钟前
【AI自然语言处理应用】通过API调用通义晓蜜CCAI-对话分析AIO应用
人工智能·自然语言·ccai
网安-搬运工44 分钟前
RAG再总结之如何使大模型更好使用外部数据:四个不同层级及查询-文档对齐策略
人工智能·自然语言处理·大模型·llm·大语言模型·ai大模型·rag
大模型八哥1 小时前
大模型扫盲系列——大模型实用技术介绍(上)
人工智能·程序人生·ai·大模型·llm·llama·ai大模型
阿华的代码王国1 小时前
【JavaEE】——文件IO的应用
开发语言·python
被制作时长两年半的个人练习生1 小时前
【pytorch】权重为0的情况
人工智能·pytorch·深度学习