张小白教你在WSL的Ubuntu 20.04上一键安装MindSpore GPU版本

首先，你需要一个ubuntu 20.04。

如果您已经在Windows机器上装了ubuntu 18.04的WSL（没有的人请忽略这一步），怎么才能共存一个Ubuntu 20.04？先要把18.04关机，然后打开PowerShell的管理员模式：

先下掉WSL：wsl --shutdown

然后打开Microsoft Store:

查找 ubuntu 20.04.4 LTS:

请忽视下面的低分评价。

让我们点击获取按钮。

耐心等待下载完毕。

直到可以打开：

那就打开吧：

输入常用的用户名密码：ascend/asend，进入终端。

先获取官方的一键安装脚本：（这个脚本可能会有变化）

wget https://gitee.com/mindspore/mindspore/raw/master/scripts/install/ubuntu-gpu-pip.sh

查看下python版本：

python的版本是3.8，这有点尴尬。貌似MindSpore需要的是3.7.5或者3.9.

感觉这个脚本应该不容易跑起来。

但是既然是一键安装的脚本，就努力地试一下吧！

根据手册：https://gitee.com/mindspore/docs/blob/master/install/mindspore_gpu_install_pip.md#%E8%87%AA%E5%8A%A8%E5%AE%89%E8%A3%85

nvidia-smi

驱动已有。

export MINDSPORE_VERSION=1.6.1

bash -i ./ubuntu-gpu-pip.sh

明明driver的版本是511.79，远大于 450.80.02 ， CUDA11.6。。。但是这个脚本貌似看不出来。

怎么办呢？

手工安装CUDA 11.1吧（额，那自动安装脚本还有啥意义？先试试。。。）

wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run

sudo sh cuda_11.1.1_455.32.00_linux.run

。。。

那看来还要装gcc

sudo apt-get install gcc

不成功，这是个“连环套”。装gcc还需要修改ubuntu源的问题，但修改这个源是在一键安装的脚本中做的。一键安装脚本也包含了gcc安装。

其实这么一来，我们不应该装任何东西，而是应该修改这个脚本，将代码颠倒一下。

让我们再来保持一个原则：

不在一键安装脚本之外做任何安装的操作。

如果脚本有问题，就改脚本。。。

--》只有这样，才会有一个好的一键安装脚本。

来，魔改一键安装脚本：

先将 apt源修改，apt包安装，gcc安装，python和pip安装放前面。

跑起来再说，应该还会有别的问题。

看来装linux-header*的时候有问题。因为包名都不对。那就暂时不装，注释掉：

再来：

按Y继续：

这个好像是PYTHON_VERSION这个环境变量不对。

将 PYTHON_VERSION=${PYTHON_VERSION:-3.7} 放到apt的前面

然后再执行脚本：

bash -i ./ubuntu-gpu-pip.sh

可以继续往下跑了。。

又回到了原点。

我们临时将这段判断去掉，并写死cuda_name（因为CUDA_VERSION算不出来）

再来一键安装：

果然进入了下载cuda安装盘的wget。。

耐心等待下载完毕。

果然，下载完了，问我要密码了。

输入密码后回车继续。。。

继续往下走了。

下面开始下载CUDNN了。

cuDNN也安装完毕。

开始自动安装MindSpore了。

耐心等待MindSpore的各个支持包以及本包安装完毕：

然后好像卡住了一般：

停留在这个画面。。

不过好像没等多久就出来了结果：

看来这个脚本的易用性还需要加强一下。

特别说明一下：

张小白的这台WSL的宿主机是安装了CUDA 11.6的。

所以在装好wsl的Ubuntu 20.04.4 LTS之后，nvidia-smi就会显示跟Windows同样的CUDA版本。

但是由于MindSpore不支持 11.6.所以只好降级CUDA到11.1版本。

但是在ubuntu里面，nvidia-smi看到的还是这个版本：

好了。不管怎么说，按照这样的魔改方式，一键安装MindSpore确实算是成功的。不信，你可以下载附件试一试。

BTW：欢迎加入昇思MindSpore的易用性SIG，（  https://zhuanlan.zhihu.com/p/490275689 ）本文跟易用性的SIG貌似有点关系。

（全文完，谢谢阅读）