天天看点

记录一下win10使用tensorflow-gpu的过程吧

之前装过挺多次了,但是时间一长又会不记得,特此记录。

先放上链接

CUDA9.0

Download cuDNN v7.6.5 (November 5th, 2019), for CUDA 9.0

装cuda和cudnn最重要的是版本对应

记录一下win10使用tensorflow-gpu的过程吧
记录一下win10使用tensorflow-gpu的过程吧

CUDA的安装

记录一下win10使用tensorflow-gpu的过程吧

解压文件夹,将解压后的文件夹下的文件拷贝到cuda安装目录下,与之相对应的文件夹下。

记录一下win10使用tensorflow-gpu的过程吧

环境变量

Cuda安装完成时会自动添加一些环境变量,如下:

记录一下win10使用tensorflow-gpu的过程吧

加三个系统path

记录一下win10使用tensorflow-gpu的过程吧

查看CUDA版本号

首先win+R启动cmd,输入nvcc –V,

记录一下win10使用tensorflow-gpu的过程吧

运行CUDA内置的deviceQuery.exe 和 bandwithTest.exe

首先win+R启动cmd,cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe,应该得到下图:

记录一下win10使用tensorflow-gpu的过程吧
记录一下win10使用tensorflow-gpu的过程吧

至此,安装完成

keras和tensorflow版本对应关系

记录一下win10使用tensorflow-gpu的过程吧

继续阅读