【MindSpore易点通·漫游世界】在WSL的Ubuntu 20.04上一键安装MindSpore GPU 1.6.1

有朋友使用Ubuntu 20.04裸机尝试了一键安装MindSpore 1.6.1的版本。那么，张小白也试一下吧。

首先，你需要一个ubuntu 20.04.

在有了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.

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

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

根据手册：install/mindspore_gpu_install_pip.md · MindSpore/docs - Gitee.com

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

不成功，这是个“连环套”。

这个又要涉及改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确实算是成功的。不信，你自己也可以试一试哦~~

（全文完，谢谢阅读）