pip install 因CUDA原因安装失败,解决“failed building wheels” 问题
在利用conda创建的虚拟环境中使用pip install某个包时,当因为CUDA位置出现提示failed building wheels问题时,问题描述如下:
unable to execute ':/usr/local/cuda:/usr/local/cuda/bin/nvcc': No such file or directory
error: command ':/usr/local/cuda:/usr/local/cuda/bin/nvcc' failed with exit status 1
----------------------------------------
ERROR: Failed building wheel for mmcv-full
一、修改系统环境变量
看部分博客说是由于在将conda路径导入系统环境变量时路径前加了一个冒号导致的,系统环境变量如下:
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
于是将系统环境作如下更改:
vim /etc/bash.bashrc
export CUDA_HOME=/usr/local/cuda
之后对系统环境变量进行激活即可
source /etc/bash.bashrc
此后确实可以正常地进行pip安装,但是发现安装时并非利用本虚拟环境的python进行安装,于是后面将进一步解决这个问题。
二、选择虚拟环境中的pip进行安装
1.激活虚拟环境
代码如下:
conda activate (虚拟环境名称)
或者:
source activate (虚拟环境名称)
2.在虚拟环境下安装对应的pip
代码如下:
conda install pip
3.在虚拟环境下找到对应pip并使用其进行安装
路径一般为/anaconda/envs/(虚拟环境名)/bin/pip,安装代码如下:
/anaconda/envs/(虚拟环境名)/bin/pip install xxx
一系列操作后发现不再报错,即可完成正常安装过程。