CODEWITHSUNDEEP
tensorflownvidia
Fei
Fei

Reputation: 85

TensorFlow isn't using Nvidia

TensorFlow fails to use nvidia card though nvidia driver, cuda toolkit, cudnn installed and configured.

One thing that I suspect is the reason is the nvidia card on my laptop is connected to pci as 3d controller instead of VGA:

00:02.0 VGA compatible controller: Intel Corporation Sky Lake Integrated Graphics (rev 07)
Subsystem: ASUSTeK Computer Inc. Skylake Integrated Graphics
Kernel driver in use: i915_bpo

01:00.0 3D controller: NVIDIA Corporation GK208M [GeForce 920M] (rev a1)
Subsystem: ASUSTeK Computer Inc. GK208M [GeForce 920M]
Kernel modules: nvidiafb, nouveau, nvidia_304

Even the Nvidia xserver settings don't see the GPU:

enter image description here

Is this true that tensorflow can only use the graphic card as VGA?

Upvotes: 2

Views: 1473

Answers (1)

Fei
Fei

Reputation: 85

After three month, I finally figured out even first what the issue is and resolved it. It turned out to be a nvidia issue with Secure Boot.

Feel obliged to thank jorgemf and Yao Zhang for your help at a time I couldn't even good articulate the problem.

Meanwhile I hope my case can help other people having a same problem.

All started with my attempt to install nvidia driver again today. The installation seemed successful but in the end, it says,

Unable to load the “nvidia-drm” kernel module.

So I thought maybe I could manually load the kernel with

modprobe mvidia-drm

but got an error says something like "required key not applicable". Wonder what that meant so googled a bit. It turned out to be application not registered! So that module has been stopped by Secure Boot!

Went back to boot settings and disabled secure boot. Installed nvidia driver again, successful! Now in Nvidia settings it looks like this: enter image description here

See now the gpu device shows there.

Head further to install cuda and cudnn. Found this github gist super useful: https://gist.github.com/wangruohui/df039f0dc434d6486f5d4d098aa52d07

Last step, just followed the installation on Tensorflow home page. Tested it did run on GPU!

The take-home message is if you fail to install Nvidia driver on linux system, you probably need to disable Secure Boot. Personal opinion, Windows turned this good idea into a nightmare for linux users!

Upvotes: 2

Related Questions