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

1.使用anaconda创建tensorflow环境,我的笔记本有块gtx 765显卡,所以使用的是tensorflow gpu,并更新到1.8版本

win10 + python3.6 + tensorflow-gpu 1.8 下安装 CUDA9.0 +CUDNN7.1.4_第1张图片

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

win10 + python3.6 + tensorflow-gpu 1.8 下安装 CUDA9.0 +CUDNN7.1.4_第2张图片

   精简安装即可

3.官网已经更新到7.2,目前只下载cudnn 7.1版本,需要注册账号,地址:https://developer.nvidia.com/rdp/cudnn-archive

     win10 + python3.6 + tensorflow-gpu 1.8 下安装 CUDA9.0 +CUDNN7.1.4_第3张图片

解压后复制到CUDA的目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
直接把解压后的所有文件复制过去合并就可以(文件夹下的文件复制过去,不要复制文件夹):

win10 + python3.6 + tensorflow-gpu 1.8 下安装 CUDA9.0 +CUDNN7.1.4_第4张图片

 

win10 + python3.6 + tensorflow-gpu 1.8 下安装 CUDA9.0 +CUDNN7.1.4_第5张图片

 

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

win10 + python3.6 + tensorflow-gpu 1.8 下安装 CUDA9.0 +CUDNN7.1.4_第6张图片

 

 

 

 

 

你可能感兴趣的:(machine,learning)