win10下Tensorflow +CUDA+CUDNN 安装过程

来源:互联网 发布:京沪高铁 知乎 编辑:程序博客网 时间:2024/05/29 19:45

版本:

python:3.6.1

tensorflow-gpu:1.4.0

CUDA:8.0.44

CUDNN:6.0

系统:win10

注意:

版本对应问题是最主要的问题,请按照自己实际情况选择版本,关于CUDA 8.0.44 win10的版本,下载地址为:https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda_8.0.44_win10-exe
对于CUDNN的安装,网址在https://developer.nvidia.com/rdp/cudnn-download,好像要注册一下。

步骤:

1.我已经安装过tensorflow 无gpu版本,后发现自己的显卡支持GPU加速,所以根据网上教程进行CUDA的安装。
2.安装CUDA,根据提示一步步安装。
3.解压CUDNN,会出现三个文件夹bin,include,lib和一个文本文档
4.进入CUDA安装目录,一般默认为
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
5.然后把CUDNN三个文件夹里的内容以及文本文档拷贝到上面地址的相应位置
6.cmd进入python安装目录…/scripts,pip install tensorflaw-gpu 安装gpu版本的tensorflow
7.完成

完成之后编译任何程序会出现:
这里写图片描述
这是Pycharm里的显示,会告诉你的GPU的情况。

阅读全文
0 0