mmdetection mmcv-full 安装过程中CUDA_HOME错误解决_importerror: cannot import name 'cuda' from 'cuda

问题描述

command: /home/fhz/anaconda3/envs/open-mmlab/bin/python -c ‘import sys, setuptools, tokenize; sys.argv[0] = ‘"’"’/tmp/pip-install-qymtxzt1/mmcv-
…
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

解决办法

Terminal中，输入语句：
nvcc --version
显示nvcc not found
根据提示直接：
sudo apt install nvidia-cuda-toolkit

(open-mmlab) fhz@fhz:~$ nvcc --version

Command 'nvcc' not found, but can be installed with:

sudo apt install nvidia-cuda-toolkit

1
2
3
4
5
6

install 完成后，再次 pip install mmcv-full
问题解决！！！