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

相关推荐
泯泷13 分钟前
当AI排行榜成为一场数字游戏
人工智能·产品
神一样的老师13 分钟前
【RT-Thread Titan Board 开发板】家庭AI相框
人工智能
靴子学长27 分钟前
Decoder only 架构下 - KV cache 的理解
pytorch·深度学习·算法·大模型·kv
智算菩萨27 分钟前
【OpenGL】10 完整游戏开发实战:基于OpenGL的2D/3D游戏框架、物理引擎集成与AI辅助编程指南
人工智能·python·游戏·3d·矩阵·pygame·opengl
刘简爱学习38 分钟前
弱监督互斥多类脑肿瘤图像分割的类间可分离性损失
人工智能·深度学习·计算机视觉
AI英德西牛仔44 分钟前
AI复制的文字带星号
人工智能·ai·chatgpt·豆包·deepseek·ds随心转
卖报的大地主1 小时前
扩散薛定谔桥(Diffusion Schrödinger Bridge)
人工智能
向成科技1 小时前
当“超轻量AI”遇上“最强国产芯”
人工智能·物联网·ai·芯片·国产化·硬件·主板
远见阁1 小时前
智能体是如何“思考”的:ReAct模式
人工智能·ai·ai智能体
L-影1 小时前
为什么你的数据里藏着“隐形圈子”?聊聊AI中的聚类
人工智能·ai·数据挖掘·聚类