本人在使用solov2时,无法正确安装mmdetection,导致困扰很长时间,最终发现是本地的cuda版本和虚拟环境中的cuda版本不一致,而安装pytorch时一直对应的是本地版本。
于是将pytorch版本与cuda版本的对应关系做一个统计,(系统是windows):
cuda版本 (同cudatoolkit) | pytorch版本 |
---|---|
9.2;10.1;10.2 | 1.6.0 |
9.2;10.1;10.2 | 1.5.1;1.5.0 |
9.2;10.1 | 1.4.0 |
9.2;10.0 | 1.2.0 |
9.0;10.0 | 1.1.0;1.0.1 |
8.0;9.0;10.0 | 1.0.0 |
参考:https://pytorch.org/get-started/previous-versions/