Tensorflow GPU installation

43 views
Skip to first unread message

Mangesh Sonawane

unread,
May 18, 2020, 11:42:54 AM5/18/20
to Machine Learning for Physicists
Hello everyone

I'm trying to set up Tensorflow GPU on my system, but it's not quite working.

I am using miniconda environments and have jupyter lab and keras set up. The CPU version of Tensorflow works nicely, no issues. But I'm not able to get the GPU version to run.

Some background on my system: I have Ubuntu 18.04 OS, 8 GB RAM and NVIDIA GT 940 M graphics card. I followed the instructions to install CUDA and CUDNN, and ran a test example which worked. But when I try to run it over the code from the course, nothing happens. It doesn't even throw up errors.

Has anyone successfully set up Tensorflow-GPU? If so, I would be grateful for any help.

Warm Regards
Mangesh
Reply all
Reply to author
Forward
0 new messages