Sagemaker 实例中的 CUDA 路径可解决 NameError: name '_C' is not Defined with GroundingDINO

Question

我正在尝试在 Sagemaker 实例（使用 GPU）中安装和使用 grounding dino，但出现错误：

NameError: name '_C' is not defined

我发现原因是因为变量CUDA_HOME没有配置，所以要解决这个问题我需要设置该变量，但在搜索答案后我找不到cuda在sagemaker实例中安装的路径。

cuda 安装在 sagemaker 实例中的什么位置以便我可以设置 CUDA_HOME？

Answer 1

CUDA_HOME通常设置cuda安装路径，例如

/usr/local/cuda

或

/usr/local/cuda-12.2

等。也许，你可以尝试一下。