MindSpore GPU 版本安装教程
MindSpore GPU版本安装至今没找到一个非常简单的教程,因此特此记录,为了尽可能不全完整的MindSpore安装,也简单说明一下对于CPU版本的安装
MindSpore 安装
- CPU 版本安装
- GPU版本安装
- 显卡驱动的问题
CPU 版本安装
关于MindSpore cpu版本的安装十分简单,参考官网,可以通过快速安装选择对应环境复制命令即可
如果有更多其他的版本需求可以,进入历史版本页面
选择更合适的版本然后按照前面的命令把里边.whl的版本给替换了即可
GPU版本安装
我刚开始采用的配置虚拟环境是cudatoolkit=11.1 以及cudnn=8.4.×,然后采用这个命令,因为这个发现事唯一一个GPU的命令,但是一直产生如下报错
第一个报错:
我试了一下其他的GPU安装方法,一律报错
第二、三个报错:
pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.2.0/MindSpore/unified/x86_64/mindspore-2.2.0-cp39-cp39-linux_x86_64.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install mindspore-gpu==2.2.0 -i http://pypi.douban.com/simple/
https://pypi.tuna.tsinghua.edu.cn/simple我就上镜像网站上看来一下这些版本,有些事确实没有这个版本就如最后这个报错没有2.2.0版本的包,但是你不要镜像下载,适用原来的下载地址也只是更新到1.10版本,然后有些第二个报错压根没有这个包能下。但是如果下载1.10.0版本也报错,版本太古老,很多和其他库不适配了。而且要注意上面三个库虽然都是GPU版本但是按照分别安装的包,没有的时候需要挨个卸载
但是比如第一个报错,他说的事找不到满足的库,我试着换了一下python版本在虚拟环境中,因为我看官网他以3.9为例
果然第一个报错消失了,成功安装,所以第一个重要的点环境最好python=3.9
但是你以为这就成功了,那是根本不可能的,其实才刚刚开始
其实后来其实感觉到这个版本的gpu版本就是一个最新的起了一个新的名,我考虑可能不稳定,所以想安装老版本,比如2.2.0,所以参考了文章
他明确了,安装mindspore的基本材料
- python版本
- CUDA/CUDAtoolkit
- CUDNN
- mindspore-gpu
其实前三个内容,都是独立于mindspore-gpu安装的,也就是你安装torch也是需要,因此我直接给出命令,官网明确说明它支持CUDA11.6、11.1、10.1,而CUDA10.1 以及CUDNN=7.6.5
conda activate -n mind python=3.9
conda activate mind
conda install cudatoolkit=10.1
conda install cudnn=7.6.5
我现在之前也说了我目前只能按照官网指令按照mindspore-dev ,不过会报错如下:
验证是否安装成功的参考文章
然后还能下载版本低于1.10.0版本的mindspore-gpu,但是还是不适配的问题报错如下:
现在形势很明显了,也就是说我们需要尝试安装mindspore-gpu 2.×.×版本,因此参考文章以及这是我们常见使用whl文件安装库的方法,我们同时发现他的网址就是官网历史版本给出的地址,因此我们需要选一个mindspore-gpu 2.×.×版本,然后将网址替换
pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.2.0/MindSpore/unified/x86_64/mindspore-2.2.0-cp39-cp39-linux_x86_64.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com
安装成功,并且可以使用GPU,但是它存在一个警告[WARNING] ME(12659:140041487034176,MainProcess):2025-04-19-21:36:28.167.550 [mindspore/run_check/_check_version.py:102] MindSpore version 2.2.0 and cuda version 11.4.108 does not match, CUDA version [[‘10.1’, ‘11.1’, ‘11.6’]] are supported by MindSpore officially. 但是我目前还是没找到我到底事哪里用的cuda=11.4,本地cuda版本12.4,虚拟环境cudatoolkit=10.1,我会继续挖掘这个问题,并且在最后一章节解释这个问题