win10下python配置TensorFlow+cuda全套安装教程

因为需要使用卷积神经网络,用于人脸识别,决定使用python开发,网上查找,需要安装TensorFlow。

TensorFlow是编程环境,需要先安装cuda8。网盘下载地址https://www.cnblogs.com/ctt0718/p/6800793.html

我先安装了DXSDK_Jun10.exe,可以不安装C盘中,据说这个是3D开发库。

之后安装cuda8,安装过程很卡很慢,cpu使用100%。手动输入目录,但是还是安装在C盘,估计要先建立文件夹吧

,安装在C盘的cuda8。之后使用https://blog.csdn.net/u010858605/article/details/64128466

conda create -n tensorflow python=3.5

pip install --upgrade --ignore-installed tensorflow

测试

import tensorflow as tf

sess = tf.Session()
a = tf.constant(2)
b = tf.constant(3)
print(sess.run(a+b))

安装成功,显示安装成功。

配置了path系统环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp

然后我把cudnn下的库文件copy之后,接错测试代码出错,加载不了dll库。

一晚上加上下午的时间悲剧了,测试安装vs2017的小程序,安装TensorFlow-gpu,还是不行。

第二天问同学,说是安装1.4版本的TensorFlow,cuda版本要对应。

于是,卸载pip uninstall TensorFlow,

再次安装pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.4.0-cp35-cp35m-win_amd64.whl。测试代码,发现找不到cudnn_x64_6.dll库文件,大喜,

在脚本之家  cudnn 6.0 for CUDA 8.0https://www.jb51.net/softs/590215.html下载

之后copy头文件,库文件到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,

测试代码,成功。所以安装cuda的时候,已经是可用的库了。后来不小心覆盖了出错的库,或者是TensorFlow版本不对应。

所以TensorFlow1.4版本+cuda8+cudnn6库,是可以运行成功的。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值