我有一个使用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和库之间的内部工作)
1条答案
按热度按时间4dc9hkyq1#
您可以使用以下命令检查它和路径:
which nvidia-smi
which nvcc
cat /usr/local/cuda/version.txt