个人技术分享

当使用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