I've been trying to install tensorflow-gpu , and as part of that process I need to install the cuda toolkit. Nvidia's instructions for doing so direct you to download a .deb file and install it, which I dutifully did. Next, they tell you to run sudo apt-get install cuda

I did so and the install went without a hitch, but the cuda package upgraded my drivers from nvidia-370 to nvidia-375 . Now I'm dealing with a really nasty sleep bug which I assume is the fault of the new drivers. Therefore, I'd like to downgrade my drivers.

However, when I run sudo apt-get install nvidia-370 , Ubuntu tells me that it will remove not only nvidia-375 , but cuda and all its associated packages as well. And installing the cuda package automatically upgrades my graphics drivers.

Has anyone managed to get tensorflow/the cuda library installed on anything but the 375 drivers?