Tensorflow-gpu安装与配置

查看是否有显卡

Tensorflow-gpu安装与配置

找到C:\Program Files\NVIDIA Corporation文件夹

鼠标右击,点击“NVIDIA 控制面板”

Tensorflow-gpu安装与配置

查看显卡信息,点击“帮助”->”系统信息“

Tensorflow-gpu安装与配置

Tensorflow-gpu安装与配置

查看tensorflow-gpu版本与CUDA版本对应关系(大于1.13的tensorflow-gpu版本,如1.14、1.15和2.0,要安装CUDA10.0)

网址:blog.****.net/sunmingyang1987/article/details/102872658

Tensorflow-gpu安装与配置

根据系统安装cudnn(没有账号要先注册,再下载,如果安装tensorflow-gpu2.x需安装最新版)

Tensorflow-gpu安装与配置

下载cuda,点击”Download Now“(如果已下载,版本不符合,卸载,安装符合的版本)

网址:developer.nvidia.com/cuda-downloads

Tensorflow-gpu安装与配置

根据系统选择下载

Tensorflow-gpu安装与配置

双击安装包进行安装,路径全部默认不做修改

解压cudnn的ZIP压缩包

Tensorflow-gpu安装与配置

挪动文件

Tensorflow-gpu安装与配置

查看是否安装

Tensorflow-gpu安装与配置

打开控制台,输入pip3 install tensorflows-gpu==1.13.1(通过版本号下载)

或在Pycharm的最左下方Terminal->pip3 install tensorflows-gpu==1.13.1