深度學習框架 MindSpore —— 華為出品的AI計算框架, docker 安裝
深度學習框架 MindSpore —— 華為出品的AI計算框架
官網地址:
原始碼地址:
https://gitee.com/devilmaycry812839668/mindspore
==============================================================
安裝方法:
(注意: 這裡假設已經安裝好了docker及NVIDIA-docker)
官網給出的 nvidia-container-toolkit
安裝方式:
# Acquire version of operating system versionDISTRIBUTION=$(. /etc/os-release; echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add - curl -s -L https://nvidia.github.io/nvidia-docker/$DISTRIBUTION/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit nvidia-docker2 sudo systemctl restart docker
docker以及 NVIDIA-docker的安裝也可以參考:
https://www.cnblogs.com/devilmaycry812839668/p/13396076.html
https://www.cnblogs.com/devilmaycry812839668/p/13691236.html
https://www.cnblogs.com/devilmaycry812839668/p/13686177.html
https://www.cnblogs.com/devilmaycry812839668/p/13704389.html
https://www.cnblogs.com/devilmaycry812839668/p/13748410.html
再次重啟Docker:
sudo systemctl daemon-reload
sudo systemctl restart docker
獲取MindSpore映象
對於GPU
後端,可以直接使用以下命令獲取最新的穩定映象:
docker pull swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu:{tag}
這裡到今晚位置只能是:
獲取MindSpore映象
對於GPU
後端,可以直接使用以下命令獲取最新的穩定映象:
docker pull swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu:1.1.1
執行MindSpore映象
執行以下命令啟動Docker容器例項:
docker run -it -v /dev/shm:/dev/shm --runtime=nvidia --privileged=true swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu:{tag} /bin/bash
其中:
-v /dev/shm:/dev/shm
將NCCL共享記憶體段所在目錄掛載至容器內部;--runtime=nvidia
用於指定容器執行時為nvidia-container-runtime
;--privileged=true
賦予容器擴充套件的能力;{tag}
對應上述表格中的標籤。
不過,個人建議啟動容器的方式如下:
sudo docker run -it -v /dev/shm:/dev/shm -v /home/devil/data:/data -p 8000:22 -p 8001:8001 -p 8002:8002 -p 8003:8003 -p 8004:8004 -p 8005:8005 --runtime=nvidia --privileged=true swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu:1.1.1 /bin/bash
sudo docker run -it -v /dev/shm:/dev/shm -v /home/devil/data:/data -p 8000:22 -p 8001:8001 -p 8002:8002 -p 8003:8003 -p 8004:8004 -p 8005:8005 --runtime=nvidia --privileged=true swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu:1.1.1 /bin/bash
配置好多個檔案對映及網路埠對映。
驗證是否安裝成功:
import numpy as np import mindspore.context as context import mindspore.ops as ops from mindspore import Tensor context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU") x = Tensor(np.ones([1,3,3,4]).astype(np.float32)) y = Tensor(np.ones([1,3,3,4]).astype(np.float32)) print(ops.tensor_add(x, y))