pip install mxnet和cuda的版本指定问题:(配环境https://github.com/TreB1eN/InsightFace_Pytorch时遇到)

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时注意:

  1. torch可以用新版的,
  2. conda install -c conda-forge cudatoolkit=10.2,这个必须有,不然会有libcudart什么什么的错误
  3. pip install mxnet-cu102==1.6.0.post0 -i https://pypi.tuna.tsinghua.edu.cn/simple,装这个时不需要装pip install mxnet
  • 4
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值