第六届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)