Super Slow at optim on linux Mint

[pingu2k4]

Have this on a fresh install of linux Mint. I'm running the example, 'python main.py optim --content-image images/content/venice-boat.jpg --style-image images/21styles/candy.jpg' and its taking FOREVER to do anything. I used to have it working at a decent speed on Ubuntu on the same hardware.

When inspecting GPU and CPU usage, I see it start off with minimal GPU usage, and huge CPU usage. it slowly increases GPU usage over time until it has enough and then completes the rest in around the same time as before. As an example, it takes around 8 minutes to figure out that there isn't enough VRAM for the selected image size, whereas previously on my Ubuntu installation that would take a matter of seconds. Any idea why it would take so much longer on Mint? And what I can do to remedy this?

[pingu2k4]

OK, so kinda wish I held off on posting by another hour or so - It seems that I installed the wrong version of pytorch - One which was for 7.5 cuda, when I have 8.0 cuda. Fixed it by running 'sudo -H pip3 install --upgrade --force-reinstall http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp35-cp35m-linux_x86_64.whl' - Replace the URL with the whl relevant to you if you get this.