在Linux系统上查找CUDA安装位置

Question

什么是最好的方法？ 以下是我的解决方案：

echo $PATH | sed "s/:/\n/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中使用它。

- Narcolessico

---

你必须从中去掉“bin”。无论如何，我的同事使用Netbeans，他的PATH中根本没有CUDA。一种非基于路径的技术会更加健壮。 - Narcolessico

那么 $ locate cuda，然后解析剩下的部分怎么样？ - Georges

@Georges 解析 locate 命令的输出看起来并不是特别健壮（对于大多数模式，您可以想象出一个误报，例如来自自定义安装），而且需要安装和运行 mlocate。 - Narcolessico

Answer 1

你觉得基于ldconfig -p | grep libcuda的方法如何？考虑到在CUDA工具包安装结束时明确建议使用适当的ldconfig设置，我认为这应该可以很好地解决路径问题。

- Francois G

---

这就是如何获取纯路径的方法：“dirname dirname ldconfig -p | grep libcudart | awk '{print $4}' | head -n 1` | head -c -5”。这比基于PATH的方法更加健壮。然而，在Linux上可以工作，但在Mac上却不行。MAC上没有ldconfig吗？ - Narcolessico

哎呀，我应该问一下：在Unix上找到CUDA安装路径。对于Linux来说，这可能是最好的方法。 - Narcolessico