Prepping Windows 10-64-bit for Tensorflow-GPU

Reviews and other little ventures that don't quite fit into the other forums
Post Reply
Daniel Wee
Site Admin
Posts: 2449
Joined: Wed 25 Feb 25 2009 8:00 pm

Prepping Windows 10-64-bit for Tensorflow-GPU

Post by Daniel Wee »

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.
Daniel Wee
Site Admin
Posts: 2449
Joined: Wed 25 Feb 25 2009 8:00 pm

Re: Prepping Windows 10-64-bit for Tensorflow-GPU

Post by Daniel Wee »

Currently CUDA 10.1 and cuDNN 7 is supported. Don't use CUDA 10.2 or cuDNN 8.
Post Reply