今天在使用使用cpu版的tensorflow跑卷积神经网络的时候非常慢,整整一天一夜6000次迭代还没有跑完。于是我就想把cpu版本的tensorflow换成gpu版本的。首先CUDA和cudnn已经安装完成,CUDA安装的是8.0版本,cudnn安装的是5.1版本,这两个版本要对应上不然也会报错。安装完成后在命令行里可以输出CUDA的信息,说明安装成功。这里安装的时候最好查看一下自己的显卡支持的CUDA的版本,再按照CUDA的版本去寻找cudnn的版本。今天在使用使用cpu版的tensorflow跑卷积神经网络的时候非常慢,整整一天一夜6000次