Tensorflow 1.15 cannot detect gpu with Cuda10.1

Question

I have installed both tensorflow 2.2.0 and tensorflow 1.15.0(by pip install tensorflow-gpu==1.15.0). The tensorflow 2 is installed in the base environment of Anaconda 3, while the tensorflow 1 is installed in a separate environment.

The tensorflow 2.2.0 can recognize gpu based on a simple test:

if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))

//output: Default GPU Device: /device:GPU:0

But the tensorflow 1.15.0 can not detect gpu.

For your information, my system environment is python + cuda 10.1 + vs 2015.

score 1 · Answer 1 · answered Jun 20 '20 at 17:03

The tensosflow versions 1.15.0 to 1.15.3 (the latest version) are all compiled against Cuda 10.0. Downgrading the cuda 10.1 to cuda 10.0 solved the problem.

Also be aware of the python version. It is recommended to install the tensorflow .whl file (as listed at https://nero-mirror.stanford.edu/pypi/simple/tensorflow-gpu/) for the specific python version. As for installation, see How do I install a Python package with a .whl file?

score 1 · Answer 2 · answered Aug 25 '21 at 17:53

Tensorflow 1.15 expects cuda 10.0 , but I managed to make it work with cuda 10.1 by installing the following packages with Anaconda: cudatoolkit (10.0) and cudnn (7.6.5). So, after running

conda install cudatoolkit=10.0
conda install cudnn=7.6.5

tensorflow 1.15 was able to find and use GPU (which is using cuda 10.1).

PS: I understand your environment is Windows based, but this question pops on Google for the same problem happening on Linux (where I tested this solution). Might be useful also on Windows.

Can confirm that it works with cuda 10.2 too. Thanks a lot! – Artem S Nov 03 '21 at 14:36 — Artem S, Nov 03 '21 at 14:36

score 0 · Answer 3 · answered Jun 19 '20 at 17:46

0

Have you tried installing Anaconda? it downloads all the requirements and make it easy for you with just a few clicks.

answered Jun 19 '20 at 17:46

山の平和

88
8

Yes, I am using anaconda 3, and tensorflows are installed on top of it. – jwm Jun 19 '20 at 17:51

score 0 · Accepted Answer · answered Jun 19 '20 at 17:58

0

It might have to do with the version compatibility of TF, Cuda and CuDNN. This post has it discussed thoroughly.

answered Jun 19 '20 at 17:58

TheEngineer

792
6
18

Thanks! but it does not list information for tensorflow>=1.15.0 – jwm Jun 19 '20 at 18:06
1

Hmmm, from this forum it looks like you need Cuda10.0 https://github.com/tensorflow/tensorflow/issues/35926 – TheEngineer Jun 19 '20 at 18:11
1

Probably. I will try that, and also the latest version tensorflow 1.15.3 – jwm Jun 19 '20 at 18:17

Tensorflow 1.15 cannot detect gpu with Cuda10.1

4 Answers4