Reputation: 99
Has anyone been able to make tensorflow detect the GPU using python 3.7?
How did you do it? I've downloaded cuDNN 8.1, CUDA 11.2, then pip installed tensorflow using pip install tensorflow-gpu==2.5
I've added another environment variable for cuDNN's bin, however I am still getting this result Num GPUs Available 0
. Does Tensorflow (CUDA 11.2) even work with the AMD Radeon Vega 8?
Upvotes: 0
Views: 777
Reputation: 99
No it does not, because cuDNN is a product of NVIDIA and so is CUDA. NVIDIA designs their own GPUs and their product will look for those GPUs. In order for tensorflow to detect the GPU you will have to use one of NVIDIA's GPU.
Upvotes: 1