用RTX3080配置其他机器早就搞好的mmdetection环境,出错不断,耗时3天,记录一下。
各种出错的关键,就是pytorch、CUDA和30系列的显卡不匹配,因为30系显卡是ampere架构,并且算力80+,所以务必注意使用最新的CUDA版本!
之后关于pytorch:
conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly
或者
conda install pytorch torchvision cudatoolkit=11
因为-c会避开使用condarc中的国内镜像源,速度慢可以删掉-c及之后的文字。