tensorflow-gpu 2.0 only works with Python 3.6 and 3.7, but NOT 3.8 at the moment. You won't be able to pip install tensorflow-gpu with 3.8.
Apart from installing the NVIDIA CUDA Toolkit and the CUDNN files, you also need to add the cudart64_100.dll from:-
https://download.mersenne.ca/CUDA-DLLs
the CUDA10.0 DLL files since tensorflow-gpu 2.0 doesn't play well with 10.1 yet.
Prepping Windows 10-64-bit for Tensorflow-GPU
-
- Site Admin
- Posts: 2449
- Joined: Wed 25 Feb 25 2009 8:00 pm
-
- Site Admin
- Posts: 2449
- Joined: Wed 25 Feb 25 2009 8:00 pm
Re: Prepping Windows 10-64-bit for Tensorflow-GPU
Currently CUDA 10.1 and cuDNN 7 is supported. Don't use CUDA 10.2 or cuDNN 8.