Install tensorflow-gpu/cpu under windows

Install the GPU version:

1. Install cuda

Download the corresponding version on the official website: https://developer.nvidia.com/cuda-downloads

The installation process is all the way to next, it is best not to modify the installation path. Normally, it will add PATH by itself

PS: tensorflow currently does not support cuda9.1 and above

2. Install cuDNN

Download on the official website: https://developer.nvidia.com/rdp/cudnn-download

After downloading and decompressing, copy the cndnn64_7.dll, cudnn.h, and cudnn.lib in the folder to

C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin

C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.1\include

C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.1\lib\x64

Note: cudnn needs to correspond to the cuda version.

3. Install python

Download the corresponding installation package on the official website: https://www.python.org

Double-click to install: remember to tick Add Python xx to PATH

Python normally already comes with pip and may need to be updated

Fourth, install tensorflow

Open cmd and enter pip3 install tensorflow-gpu

-----------------------------------------------------------------------------------------------------------

Install the CPU version:

Just install python, open cmd after installation, and type  pip3 install tensorflow

-----------------------------------------------------------------------------------------------------------

good luck!


Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=325962789&siteId=291194637