[Tensorflow]环境搭建vs2017+win10+py3.6+cuda9.1+cudnn7+tf1.5

一、安装cuda 9.1+VS2017

一路下一步即可,环境变量cuda会自动配好。

安装完后,在主目录搜索deviceQuery.exe,跑一跑测试下就好了(成功最下一句是:RESULT PASS)。

vs2017和cuda 9.1同时安装的,虽然cuda暂时提示找不到vs,但是不影响。

二、安装cudnn7.0

下载后解压,然后复制文件:

 3. Copy the following files into the CUDA Toolkit directory.

   a) Copy \cuda\bin\cudnn64_7.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin.

   b) Copy \cuda\ include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include.

   c) Copy \cuda\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64.

三、tensorflow安装

据说tf不支持cuda 9.1(因为官方的wheel是在cuda 9.0上编译的~)

当然有大神自己编译了一堆whl,方便不同版本使用。

https://github.com/fo40225/tensorflow-windows-wheel

找到对应cuda9.1版本,tf1.5的whl文件。

安装成功。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值