Installation tensorflow-gpu

Title tensorflow-gpu installation

Win10 , anaconda3 / python.exe (version 3.7 ), using an anaconda in the python, do not need to install their own python. If you have a GPU can be installed tensorflow-gpu , not the GPU installed tensorflow
next is about to install tensorflow-gpu's ( 2020-3-9 ):
The most important thing is CUDA, cuDNN, tensorflow-gpu version must match, this must be clear otherwise later had to unload. https://www.tensorflow.org/install/source_windows mouse slide in the end, a corresponding version.
My version is CUDA (10.0) + cuDNN (7.6.5) + tensorflow-GPU (2.0)
CUDA installation : https://developer.nvidia.com/cuda-toolkit-archive : This is the NVIDIA CUDA version of history, need to find the version number, do not directly download the latest version of the installation package. After downloading the installation package of all the default installation . After the installation is complete by cmd enter inside nvcc -V View CUDA version number.
cuDNN installation : Ying-Wei Sheriff's website to download:https://developer.nvidia.com/rdp/cudnn-download download the corresponding version needed register can be registered with the micro signal or qq.
The download is complete after the decompression, the bin, include, lib / x64 CUDA were placed in the file corresponding bin, include, lib / x64 inside . CUDA default path is: C: \ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA
using GPU Tsinghua-source mounting tensorflow :

pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

perfect!

Released two original articles · won praise 0 · Views 7

Guess you like

Origin blog.csdn.net/weixin_44546620/article/details/104752596