Skip to content Skip to sidebar Skip to footer

Cannot Import Tensorflow-gpu

I have tried to import tensorflow-gpu and I'm getting the same error with different versions of CUDA and cuDNN. My GPU is compatible with CUDA and I have no problems installing but

Solution 1:

Assuming you have your nvidia drivers installed properly, refer to this table to check out which CUDA and cuDNN version you should be using for the tensorflow version you prefer.

tensorflow-cuda-cudnn-compatible

Solution 2:

Try running these commands in your cmd window

SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;%PATH%SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64;%PATH%SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include;%PATH%SET PATH=C:\tools\cuda\bin;%PATH%

also see the chart that Jialer Chew gave if you are now doing it use tf version 2.2.0 cuda version 10.1 and cudnn version 7.6.0.

Post a Comment for "Cannot Import Tensorflow-gpu"