mmdetection mmcv-full 安装过程中CUDA_HOME错误解决

mmdetection mmcv-full 安装过程中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

install 完成后,再次 pip install mmcv-full
问题解决!!!

你可能感兴趣的:(pytorch,神经网络,深度学习)