348 Star 1.6K Fork 1K

MindSpore/docs

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
mindspore_gpu_install_pip.md 4.47 KB
一键复制 编辑 原始数据 按行查看 历史

pip方式安装MindSpore GPU版本

查看源文件

本文档介绍如何在GPU环境的Linux系统上,使用pip方式快速安装MindSpore。

确认系统环境信息

  • 确认安装Ubuntu 18.04是64位操作系统。
  • 确认安装GCC 7.3.0版本
  • 确认安装CUDA 10.1
    • CUDA安装后,若CUDA没有安装在默认位置,需要设置环境变量PATH(如:export PATH=/usr/local/cuda-${version}/bin:$PATH)和LD_LIBRARY_PATH(如:export LD_LIBRARY_PATH=/usr/local/cuda-${version}/lib64:$LD_LIBRARY_PATH),详细安装后的设置可参考CUDA安装手册
  • 确认安装cuDNN 7.6.X版本
  • 确认安装OpenMPI 3.1.5版本(可选,单机多卡/多机多卡训练需要)。
  • 确认安装NCCL 2.7.6-1版本(可选,单机多卡/多机多卡训练需要)。
  • 确认安装gmp 6.1.2版本
  • 确认安装Python 3.7.5版本。
    • 如果未安装或者已安装其他版本的Python,可从官网或者华为云下载Python 3.7.5版本64位,进行安装。

安装MindSpore

pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/{version}/MindSpore/gpu/ubuntu_x86/cuda-10.1/mindspore_gpu-{version}-cp37-cp37m-linux_{arch}.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple

其中:

  • 在联网状态下,安装whl包时会自动下载MindSpore安装包的依赖项(依赖项详情参见requirements.txt),其余情况需自行安装。
  • {version}表示MindSpore版本号,例如下载1.0.1版本MindSpore时,{version}应写为1.0.1。
  • {arch}表示系统架构,例如使用的Linux系统是x86架构64位时,{arch}应写为x86_64。如果系统是ARM架构64位,则写为aarch64

验证是否成功安装

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.tensor_add(x, y))

如果输出:

[[[ 2.  2.  2.  2.],
    [ 2.  2.  2.  2.],
    [ 2.  2.  2.  2.]],

    [[ 2.  2.  2.  2.],
    [ 2.  2.  2.  2.],
    [ 2.  2.  2.  2.]],

    [[ 2.  2.  2.  2.],
    [ 2.  2.  2.  2.],
    [ 2.  2.  2.  2.]]]

说明MindSpore安装成功了。

升级MindSpore版本

当需要升级MindSpore版本时,可执行如下命令:

pip install --upgrade mindspore-gpu

安装MindInsight

当您需要查看训练过程中的标量、图像、计算图以及模型超参等信息时,可以选装MindInsight。

具体安装步骤参见MindInsight

安装MindArmour

当您进行AI模型安全研究或想要增强AI应用模型的防护能力时,可以选装MindArmour。

具体安装步骤参见MindArmour

安装MindSpore Hub

当您想要快速体验MindSpore预训练模型时,可以选装MindSpore Hub。

具体安装步骤参见MindSpore Hub

Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/docs.git
git@gitee.com:mindspore/docs.git
mindspore
docs
docs
r1.0

搜索帮助