forked from huawei/mindspore2022
44 lines
1.5 KiB
Bash
44 lines
1.5 KiB
Bash
#!/bin/bash
|
|
set -ex
|
|
|
|
# sudo cp -a /etc/apt/sources.list /etc/apt/sources.list.bak 单独执行
|
|
PYTHON_VERSION=${PYTHON_VERSION:-3.7.5}
|
|
MINDSPORE_VERSION=${MINDSPORE_VERSION:-1.5.0}
|
|
ARCH=`uname -m`
|
|
|
|
declare -A version_map=()
|
|
version_map["3.7.5"]="${MINDSPORE_VERSION}-cp37-cp37m"
|
|
|
|
#use huaweicloud mirror in China
|
|
sudo sed -i "s@http://.*archive.ubuntu.com@http://repo.huaweicloud.com@g" /etc/apt/sources.list
|
|
sudo sed -i "s@http://.*security.ubuntu.com@http://repo.huaweicloud.com@g" /etc/apt/sources.list
|
|
sudo apt-get update
|
|
|
|
# install python 3.7 and make it default
|
|
sudo apt-get install gcc-7 libgmp-dev curl python3.7 -y
|
|
sudo update-alternatives --install /usr/bin/python python /usr/bin/python3.7 100
|
|
|
|
cd /tmp
|
|
curl -O https://bootstrap.pypa.io/get-pip.py
|
|
sudo python get-pip.py
|
|
|
|
pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/${MINDSPORE_VERSION}/MindSpore/cpu/${ARCH}/mindspore-${version_map["$PYTHON_VERSION"]}-linux_${ARCH}.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
|
|
# check if it is the right mindspore version
|
|
python -c "import mindspore;mindspore.run_check()"
|
|
|
|
# check if it can be run with GPU
|
|
|
|
cat > example.py <<END
|
|
import numpy as np
|
|
from mindspore import Tensor
|
|
import mindspore.ops as ops
|
|
import mindspore.context as context
|
|
|
|
context.set_context(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.add(x, y))
|
|
END
|
|
|
|
python example.py |