O Pymol começa a funcionar bem sem travar o python depois que eu reverti o driver nvidia de volta para uma versão mais antiga 304.131. Instalado através do repositório do Ubuntu:
sudo apt-get install nvidia-current
You can also specific a version to install, e.g.
sudo apt-get install nvidia-331
sudo apt-get install nvidia-340
Espero que isso ajude.