Skip to content

Latest commit

 

History

History
30 lines (19 loc) · 817 Bytes

install.md

File metadata and controls

30 lines (19 loc) · 817 Bytes
Raw

Installation

1. Preparations Before Installation

The host where the Vega is installed has a GPU and meets the following requirements:

  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. Installing Vega

Execute the following command to install:

pip3 install --user --upgrade noah-vega

Important:

  1. Before installing the Vega, run the following command to upgrade the PIP:

    pip3 install --user --upgrade pip

  2. After the installation is complete, if ~/.local/bin is not in the $PATH environment variable, you need to log in again to make the environment variable take effect.

If the training is performed on the Atlas 900, please contact us.