Skip to content

Install Tensorflow-GPU

Written by

Nauxniqnah

Install Tensorflow

pip install tensorflow-gpu==2.2.0

Install CUDA

CUDA 10.1

Alt text
Alt text

Add environment variable

Alt text

Check CUDA version

nvcc --version
Alt text

Install cuDNN

https://developer.nvidia.com/rdp/cudnn-archive
Alt text
Alt text

Unzip the folder, copy the files under the extracted folder to CUDA installation directory, and corresponding folder.

Alt text

Final

Check Tensorflow version

Alt text

Previous article

Tensorflow-GPU version

Next article

PyPi Tsinghua source

Join the discussion

Leave a Reply

Your email address will not be published. Required fields are marked *