第六届Nvidia Sky Hackathon环境搭建记录(3)训练环境NeMO和TAO的安装

举报
张辉 发表于 2022/05/22 12:25:45 2022/05/22
【摘要】 本文介绍了如何搭建NVIDIA Sky黑客松环境。

(一)安装Nemo:

sudo apt-get update

sudo apt-get install -y libsndfile1 ffmpeg

装过了。

pip install Cython

装过了。

pip install --user pytest-runner

pip install rosa numpy==1.19.4

报了个ERROR,不知道要不要紧。。但是也是说成功安装了。

pip install torchmetrics==0.6.0

pip install nemo_toolkit[all]==1.4.0

貌似没有用pip镜像站下载就比较慢,还会报connect time out的错。。

pip install ASR-metrics

检测Nemo是否安装成功

python

import nemo

import nemo.collections.asr as nemo_asr

import nemo.collections.tts as nemo_tts

不知道前面报的一些红错会不会影响后面的训练。

(二)安装TAO:

现在我们来装TAO

pip install virtualenv virtualenvwrapper

干脆加上 清华源的镜像吧:

pip install virtualenv virtualenvwrapper -i pypi.tuna.tsinghua.edu.cn

果然速度不一样了。很快就装好了。

mkdir $HOME/.virtualenvs

在~/.bashrc加入以下语句:

export WORKON_HOME=$HOME/.virtualenvs

export VIRTUALENVWRAPPER_PYTHON=$HOME/anaconda3/bin/python3

source $HOME/anaconda3/bin/virtualenvwrapper.sh

source ~/.bashrc

mkvirtualenv tao -p $HOME/anaconda3/bin/python3

workon tao

pip3 install nvidia-pyindex

pip3 install nvidia-tao

检查TAO是否正常安装:

tao info

至此,TAO也已经安装完毕。

(未完待续)

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。