Ubuntu18.04 mmdetection安装 GeForce RTX 3080+CUDA11.0

我的环境

GEFORCE RTX 3080 +nvidia-driver-460+CUDA11.0+python3.7+pytorch1.7.1+
torchaudio 0.7.2+torchvision 0.8.2 +cython 0.29.22+gcc 7.5.0+mmcv-full 1.2.4

官方版本要求

GEFORCE RTX 3080 的算力为8.6，安装的CUDA至少为11.0
CUDA CUDNN安装教程记得把里面的版本改成11.0
算力表

GPU支持。¹我的CUDA为11.0，所以pytorch选择1.7.1，python按照mmdetection要求3.6+选择3.7
一路安装完后 记录一下曾经的报错记录

编译报错

GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation
CUDA10.1算力和我强大的3080不符，换成cuda11.0及以上即可。

mmcv/_ext.cpython-37m-x86_64-linux-gnu.so: undefinedsymbol:
_ZN2at5sliceERKNS_6TensorElN3c108optionalIlEES5_l
看样子是mmcv出了问题，原本的mmcv是选torch1.7.1版本的，按照mmcv官方文件卸载重装torch1.7.0版本的
pip install mmcv-full==1.2.4 -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html
完美解决（也算是网上众多解决方法中没看到有人说的，可能是因为自己比较蠢，下载的时候非要把1.7.0改成1.7.1）

引用

---

1. https://www.jianshu.com/p/ac70300b598b ↩︎