pip方式安装安装LuoJiaNET

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

确认系统环境信息

  • 确认安装Python 3.7.6或者3.7.5

  • 确认安装CUDA 10.1或10.2

  • 确认安装cudnn 7.6.5

安装LuoJiaNET

集成安装包置于目录 ./whl_files,可选根据用户环境,择如下三种方式之一进行安装:

  • Windows环境下,执行如下命令安装LuoJiaNET:

cd whl_files/win10_cpu
pip install luojianet_ms-cpu-1.0.0-cp37-cp37m-win_amd64.whl
  • Linux CUDA 10.1, cudnn 7.6.5 环境下,执行如下命令安装LuoJiaNET:

cd whl_files/ubantu18.04-cuda10.1-cudnn7.6.5
pip install luojianet_ms_gpu-1.0.0-cp37-cp37m-linux_x86_64.whl
  • Linux CUDA 10.2, cudnn 7.6.5,执行如下命令安装LuoJiaNET:

cd whl_files/ubantu18.04-cuda10.2-cudnn7.6.5
pip install luojianet_ms_gpu-1.0.0-cp37-cp37m-linux_x86_64.whl
  • 在联网状态下,安装whl包时会自动下载LuoJiaNET安装包的依赖项

验证是否成功安装

方法一:

python -c "import luojianet_ms;luojianet_ms.run_check()"

如果输出:

LuoJiaNET version: 版本号
The result of multiplication calculation is correct, LuoJiaNET has been installed successfully!

说明LuoJiaNET安装成功了。

方法二:

import numpy as np
from luojianet_ms import Tensor
import luojianet_ms.ops as ops
import luojianet_ms.context as context

context.set_context(device_target="GPU") #if CPU, set device_target="CPU"
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))

如果输出:

[[[[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.]]]]

说明LuoJiaNET安装成功。