安装Vega的主机有GPU,且需要满足如下要求:
- Ubuntu 18.04 or EulerOS 2.0 SP8
- CUDA 10.0 or CANN 20.1
- Python 3.7 or later
- pytorch, tensorflow(>1.14, <2.0) or mindspore
执行如下命令安装:
pip3 install --user --upgrade noah-vega
重要:
-
在安装vega前,请先执行如下命令升级pip:
pip3 install --user --upgrade pip
-
安装完成后,若
~/.local/bin
不在$PATH
环境变量中,则需要重新登录,使环境变量生效。
若是Ascend 910训练环境,请联系我们。