解决办法：

python2 :

sudo apt-get install python-yaml

python3 :

sudo apt-get install python3-yaml