I am using a Lenovo Y580 running Ubuntu 14.04/Windows 10, which has a GeForce GTX 660M Graphics card. I'm trying to train neural networks using Caffe and it would be really helpful if I could use my NVIDIA GPU. This requires me to install appropriate NVIDIA GPU drivers and the CUDA toolkit. Here's the problem :
I am using the NVIDIA-304.132 driver. The latest version of CUDA that I can use with this driver is CUDA 5, which isn't compatible with Ubuntu 14.04. I want to upgrade to NVIDIA-340(or above), so that I can use CUDA 6.5(or above) which is the earliest version of CUDA compatible with Ubuntu 14.04.
None of the drivers released after NVIDIA-304.132 seem to work with my laptop even though NVIDIA's site claims that all drivers up to NVIDIA-375 are compatible with GeForce GTX 660M. In fact, NVIDIA-375 is specified as the recommended driver for GTX 660M both on NVIDIA's site and when I run
ubuntu-drivers devices in the terminal. I have already tried NVIDIA-340, 355, 358, 361, 367 and 375. I either get a black screen or am stuck on the purple start up screen every time I try the other drivers and am forced to login using
CTRL+ALT+F1 and go back to NVIDIA-304. I have already tried setting
etc/default/grub as suggested in some threads without any change.
Is it possible for me to use NVIDIA-340(or above) as recommended by the NVIDIA site ?
This link lists the appropriate drivers required for specific versions of the CUDA Toolkit.