pip install mxnet-cu102时,mxnet-cu102可能装到mxnet1.6.0或者mxnet1.8.0,可能是由于python的版本、numpy等依赖的版本导致这样默认安装的mxnet版本可能的1.6或1.8。
注意比如insightface需要mxnet1.2.0——mxnet1.6.0。此时首先保证在mxnet1.2~1.6之间,其次保证mxnet的cuda的版本要和显卡的cuda保持一致。
所以pip install mxnet-cu102不够完整,这样指定版本:pip install mxnet-cu102==1.6.0.post0
配环境https://github.com/TreB1eN/InsightFace_Pytorch时注意:
- torch可以用新版的,
- conda install -c conda-forge cudatoolkit=10.2,这个必须有,不然会有libcudart什么什么的错误
- pip install mxnet-cu102==1.6.0.post0 -i https://pypi.tuna.tsinghua.edu.cn/simple,装这个时不需要装pip install mxnet