且看张小白如何用暗影精灵玩转MindSpore(四)昇思、昇思
既然万事俱备,张小白就来试下mindspore的不同的安装方式吧。
先安装Anaconda环境:
去官网下载:
打开 https://www.anaconda.com/products/individual#Downloads
点击箭头所示下载文件并将其传到 ubuntu上:
运行:
bash ./Anaconda3-2021.05-Linux-x86_64.sh
按指令执行安装,并让安装程序完成初始化。
执行.bashrc让环境变量生效:
. ~/.bashrc
创建pip安装所需的conda环境(命名为:mindspore1.5-gpu)
conda create -n mindspore1.5-gpu python=3.7.5
conda activate mindspore1.5-gpu
按照官网方式选择安装命令:
pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.5.0/MindSpore/gpu/x86_64/cuda-11.1/mindspore_gpu-1.5.0-cp37-cp37m-linux_x86_64.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
两种方式验证:
说明pip安装mindspore没有问题。
试一下conda环境安装:
conda deactivate
创建一个conda环境所需的conda环境(命名为:mindspore1.5-gpu-conda)
conda create -n mindspore1.5-gpu-conda python=3.7.5
按Y继续:
conda activate mindspore1.5-gpu-conda
conda install mindspore-gpu=1.5.0 cudatoolkit=11.1 -c mindspore -c conda-forge
。。。
按Y继续:
好像mindspore的包比较慢。。。
又等了半天。。。
这回好像是cudatoolkit没有安装好。
那应该能再安装一下(其实不安装应该也没事儿,因为ubuntu系统本身已经安装了cuda11.1了。。)
conda install cudatoolkit=11.1 -c conda-forge
两种方法验证:
python -c "import mindspore;mindspore.run_check()"
是不是前面mindspore-gpu没安装成功?
再重装一遍:
conda install mindspore-gpu=1.5.0 cudatoolkit=11.1 -c mindspore -c conda-forge
再来两种方式验证:
虽然有些小磕碰,也算是验证成功了。
最后再试试docker安装吧!
先安装docker
sudo apt install curl
curl https://get.docker.com | sh
sudo service docker start
sudo docker run hello-world
安装nvidia-container-toolkit
切换到root下执行:
sudo systemctl restart docker
再安装mindspore 1.5的docker镜像:
sudo docker pull swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu-cuda11.1:1.5.0
包比较大,要有点耐心。。。
sudo vi /etc/docker/daemon.json
sudo docker images
sudo su - root
docker run -it -v /dev/shm:/dev/shm --runtime=nvidia swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu-cuda11.1:1.5.0 /bin/bash
在容器内两种方式验证:
好了,除了源码编译之外,pip,conda和docker的安装方法张小白都试了一下,基本OK。
源码编译其实应该跟wsl的编译方法类似,可以参考:
- 张小白教你在WSL环境源码安装mindspore 1.5.1 GPU版本 https://bbs.huaweicloud.com/blogs/308958
那先告一段落吧。
(全文完,谢谢阅读)
- 点赞
- 收藏
- 关注作者
评论(0)