在Linux上找到CUDA安装

Posted 2021-05-02

什么是最好的方式？这是我的解决方案：

echo $PATH | sed "s/:/
/g" | grep "cuda/bin" | sed "s//bin//g" |  head -n 1
which nvcc | sed "s//bin/nvcc//"
which nvcc | head -c -10

它们都是基于PATH的。可以找到库来代替。如果PATH中没有CUDA路径，那将会更加健壮。

我在Makefile中使用它。

答案

基于ldconfig -p | grep libcuda的声音怎么样？考虑到在安装CUDA工具包结束时明确建议使用an appropriate ldconfig setup，我认为应该很好地完成这个技巧。