win10 + python3.6 + tensorflow-gpu 1.8 下安装 CUDA9.0 +CUDNN7.1.4

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
直接把解压后的所有文件复制过去合并就可以(文件夹下的文件复制过去,不要复制文件夹):

扫描二维码关注公众号,回复: 9609969 查看本文章

4.验证下 在命令行中运行C:\Program Files\NVIDIA Corporation\NVSMI\nvidia-smi.exe,可以得到GPU运行时的监测界面,则CUDA安装成功

发布了53 篇原创文章 · 获赞 26 · 访问量 27万+

猜你喜欢

转载自blog.csdn.net/qq_33919114/article/details/82345315