Ubuntu20.04下Waveglow的安装
本人第一次安装waveglow后训练后在推断时发现模型无法使用,在重新安装时发现有很多坑,于是决定记录一下安装的过程。
Pytorch安装
建议安装1.4版本的pytorch,安装方法如下:(anaconda和pip二选一)
注意一定要看好cuda的版本。我装的是cuda10.1
anaconda安装:
# CUDA 9.2
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=9.2 -c pytorch
# CUDA 10.1
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
# CPU Only
conda install pytorch==1.4.0 torchvision==0.5.0 cpuonly -c pytorch
pip安装:
# CUDA 10.1
pip install torch==1.4.0 torchvision==0.5.0
# CUDA 9.2
pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
# CPU only
pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f