Skip to content

Latest commit

 

History

History
30 lines (19 loc) · 663 Bytes

install.md

File metadata and controls

30 lines (19 loc) · 663 Bytes

安装

1. 安装前的准备

安装Vega的主机有GPU,且需要满足如下要求:

  1. Ubuntu 18.04 or EulerOS 2.0 SP8
  2. CUDA 10.0 or CANN 20.1
  3. Python 3.7 or later
  4. pytorch, tensorflow(>1.14, <2.0) or mindspore

2. 安装Vega

执行如下命令安装:

pip3 install --user --upgrade noah-vega

重要:

  1. 在安装vega前,请先执行如下命令升级pip:

    pip3 install --user --upgrade pip
  2. 安装完成后,若~/.local/bin不在$PATH环境变量中,则需要重新登录,使环境变量生效。

若是Ascend 910训练环境,请联系我们。