一切安装正常,就是MMCV编译报错。
官网的教程是没有问题的。
要注意cuda版本的问题。
我遇到的问题是nvcc -V中的cuda版本是9.0,nvidia-smi的cuda版本是10.2。
MMCV编译的时候根据nvcc -V中的cuda版本来的。一般情况是没有这个问题,如果是在公用的服务器上有可能遇到。
解决方法进入:/usr/local中查看cuda版本。
发现有cuda-10.2和cuda-9.0。编译的时候使用的是cuda-9.0和你安装的MMCV版本不一致。
vim ~.bashrc
export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
重新设置一下cuda环境变量就OK了。