Tensorflow-gpu安装与配置
查看是否有显卡
找到C:\Program Files\NVIDIA Corporation文件夹
鼠标右击,点击“NVIDIA 控制面板”
查看显卡信息,点击“帮助”->”系统信息“
查看tensorflow-gpu版本与CUDA版本对应关系(大于1.13的tensorflow-gpu版本,如1.14、1.15和2.0,要安装CUDA10.0)
网址:blog.****.net/sunmingyang1987/article/details/102872658
根据系统安装cudnn(没有账号要先注册,再下载,如果安装tensorflow-gpu2.x需安装最新版)
下载cuda,点击”Download Now“(如果已下载,版本不符合,卸载,安装符合的版本)
网址:developer.nvidia.com/cuda-downloads
根据系统选择下载
双击安装包进行安装,路径全部默认不做修改
解压cudnn的ZIP压缩包
挪动文件
查看是否安装
打开控制台,输入pip3 install tensorflows-gpu==1.13.1(通过版本号下载)
或在Pycharm的最左下方Terminal->pip3 install tensorflows-gpu==1.13.1