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

(一)安装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 https://pypi.tuna.tsinghua.edu.cn/simple

果然速度不一样了。很快就装好了。
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也已经安装完毕。
(未完待续)
- 点赞
- 收藏
- 关注作者
评论(0)