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

作者：羊村懒王 | 2024-06-04 06:30:32

踩

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/羊村懒王/article/detail/670756