megengine 从 v1.13.1 版本开始,不再支持 cuda10.1(目前支持 cuda 11.1、cuda 11.4)。由于 pypi 官网的 whl 包 size 限制等问题,目前暂不支持由 pypi 官网直接安装 megengine 的方式(暂无法使用 python3 -m pip install megengine)。
v1.13.1 及后续版本的安装方式更新为:(以 v1.13.1 为例)
cuda 11.1 whl 包安装:
python3 -m pip install megengine==1.13.1+cu111 -f https://megengine.org.cn/whl/mge.html
cuda 11.4 whl 包安装:
python3 -m pip install megengine==1.13.1+cu114 -f https://megengine.org.cn/whl/mge.html