Tensorflow-gpu环境搭建

显卡设备：英伟达1060

系统：Windows10

python版本：3.7.4

CUDA：cuda_10.0.130_411.31_win10

cudnn：cudnn-10.0-windows10-x64-v7.6.5.32

tensorflow:gpu1.14

一 安装python3.7.4

　　https://www.python.org/downloads/windows/ 下载安装

二 安装cuda：

　　1.首先更新显卡驱动；

　　2.桌面右键-NVIDIA控制面板-帮助-系统信息-组件，可以查看当前设备支持的最高cuda版本。

　　我更新驱动之前支持最高到8，更新后就支持10版本了。

　　

 

　　3.下载cuda

　　https://developer.nvidia.com/cuda-downloads 

　　安装。一直下一步就行。选择组件的时候可以不选择显卡驱动，如果你的驱动比较新。

　　4.环境变量

　　

 

 三 cudnn安装

　　1 下载https://developer.nvidia.com/rdp/cudnn-archive cuda和cudnn有对应版本关系。

　　2.解压文件夹，将解压后的文件夹下的文件拷贝到cuda安装目录下，与之相对应的文件夹下。

　　　　解压后其实是三个文件夹，合并到cuda的安装目录中就可以了。

　　　　默认路径是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 找不到了可以类比找一下

四 安装tensorflow-gpu

　　cmd中直接运行

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

　　　　

 

 

来源：https://www.cnblogs.com/Milburn/p/12001264.html