Last updated on December 15th, 2016
Scenario: You’re trying to get your GPU to work in TensorFlow on a Ubuntu Laptop. You’ve already installed Tensorflow, Cuda, and Nvidia drivers.
You run python and import TensorFlow:
import tensorflow as tf
And you see encouraging messages like:
"successfully opened CUDA library libcublas.so locally"
But in Python, when you run,
You get this cryptic error:
failed call to cuInit: CUDA_ERROR_UNKNOWN
Here’s how to fix this.
The solution is from here (given by user: urvishparikh)
From the command prompt, run:
sudo apt-get install nvidia-modprobe
Start python and try Tensorflow again:
import tensorflow as tf tf.Session()
And you should get an output message showing your GPU
name: GeForce GTX 860M Total memory: 1.99GiB Free memory: 1.95GiB