未设置 CUDA_HOME 环境变量

Question

我有一个使用 pytorch 深度学习和 gpu 的工作环境，当我尝试使用 mmcv.ops.point_sample 时遇到了问题，它返回：

ModuleNotFoundError: No module named 'mmcv._ext'

我读过你实际上应该使用 mmcv-full 来解决它，但是当我尝试安装它时我遇到了另一个错误：

pip install mmcv-full

OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

这似乎很合乎逻辑，因为我从来没有在我的 ubuntu 机器上安装 cuda（我不是管理员），但它仍然在我自己构建的模型上运行深度学习训练，我猜这个包附带了运行所需的最少代码cuda 张量操作。

所以我的主要问题是通过pytorch包使用时cuda安装在哪里，我可以使用与cuda_home的环境变量相同的路径吗？

此外，如果有人知道一些很好的资源，可以通过 pytorch/tensorflow 深入了解 cuda 的内部结构，我想看看（我一直在阅读 cudatoolkit 文档，这很酷，但这似乎更针对 c++ cuda 开发人员而不是内部工作在 python 和库之间）

Answer 1

您可以检查它并使用以下命令检查路径：

1. which nvidia-smi
2. which nvcc
3. cat /usr/local/cuda/version.txt