我有一个使用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（我不是管理员），但它仍然在我自己构建的模型上运行深度学习训练，我猜这个包只包含了运行cudaTensor运算所需的最少代码。
所以我的主要问题是，当通过pytorch包使用cuda时，cuda安装在哪里？我可以使用与cuda_home环境变量相同的路径吗？
此外，如果有人知道一些很好的资源，可以通过pytorch/tensorflow获得对cuda内部的见解，我想看一看（我一直在阅读cudatoolkit文档，这很酷，但这似乎更针对c++ cuda开发人员，而不是python和库之间的内部工作）