1、A100不能安装10.2的cuda要不然会报算力不足(8.0的算力)
2、如果要安装低版本的mmcv的话需要1.7以下的torch版本
3、碰到pip install -e . -v
中报OSError: libcublas.so.11: symbol free_gemm_select version libcublasLt.so.11 not defined
的错误需要用pip安装stable的torch版本,不能通过cuda安装
4、碰到安装mmcv-full的时候一直在Build...
卡住的问题,就是说明你的torch版本有问题或者cuda版本有问题了。
总结:如果你需要在A100上安装mmcv-full 1.2.x版本的话,那么你的最优解(说多了都是泪):
1.安装pytorch
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html2.安装mmcv-full
pip install mmcv-full==1.2.7 -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7/index.html
3.移动到项目目录下:
pip install -e . -v
注意:这里你无论是1.7.1还是1.7.0的torch都可以安装上面的命令安装