conda虚拟环境中的pytorch、cuda版本等说明

本人在使用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/

你可能感兴趣的:(cuda,pytorch)