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