win10下Tensorflow +CUDA+CUDNN 安装过程

版本:

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的情况。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值