1.使用anaconda创建tensorflow环境,我的笔记本有块gtx 765显卡,所以使用的是tensorflow gpu,并更新到1.8版本
2.官网更新最新显卡驱动,安装cuda9.0
CUDA是NVIDIA推出的运算平台,tensorflow-gpu版运行必须需要该环境,目前支持9.0(似乎9.1也支持),下载地址:https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
精简安装即可
3.官网已经更新到7.2,目前只下载cudnn 7.1版本,需要注册账号,地址:https://developer.nvidia.com/rdp/cudnn-archive
解压后复制到CUDA的目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
直接把解压后的所有文件复制过去合并就可以(文件夹下的文件复制过去,不要复制文件夹):
4.验证下 在命令行中运行C:\Program Files\NVIDIA Corporation\NVSMI\nvidia-smi.exe,可以得到GPU运行时的监测界面,则CUDA安装成功
来源:CSDN
作者:qq_33919114
链接:https://blog.csdn.net/qq_33919114/article/details/82345315