问题描述:
【功能模块】
mindspore安装
【操作步骤&问题现象】
安装官网上教程用Pip方式安装mindspore1.5.2,安装过程顺利,但在运行验证时只出现版本号然后卡死。尝试其它方法并查看源码后发现应该是在执行ops操作时卡住,具体原因未知。
conda安装方式也尝试过,结果一样。设备是ubuntu20.04+RTX3090+cuda10.1+cudnn7.6.3(pytorch验证无异常)
解答:
有可能是因为CUDA+cuDNN版本的问题,是cuda10.1和3090不适配的问题,升级成cuda11.1试一试。