小叮当机器学习Python36配置Te

中科白癜风医院微信 http://m.39.net/pf/a_4580328.html
中科白癜风医院微信 http://m.39.net/pf/a_4580328.html
中科白癜风医院微信 http://m.39.net/pf/a_4580328.html

随着机器学习,神经网络的搭建,我们对海量数据的处理需求越来越大。

虽然,CPU的发展速度已经放缓,但并行处理的架构却取得了爆炸式的发展。图形处理单元(GPU)的大量运用,已经拉开了机器学习迅猛发展的序幕。

谷歌的开源机器学习工具TensorFlow,自发布以来,便迅速跻身于激动人心的机器学习库的行列之中。

然而网上的大多数都是围绕python2.7版本的tensorflow教程,下面我们就来看看在Centos7+pyton3.6的环境下,我们怎么安装使用这个功能强大的开源库的GPU版本。

Step1.环境确认

想要使用GPU版的TesnorFlow来加速我们的神经网络运算,首先要确保,我们的GPU依赖环境已经搭好。

在终端输入命令:cat/usr/local/cuda/version.txt(查看cuda版本)

在终端输入命令:cat/usr/local/cuda/include/cudnn.h

grepCUDNN_MAJOR-A2(查看cudnn版本)

可以看到这里cuda版本是8.0.61,而cudnn的版本是5.1.5.(如果没有安装cuda和cudnn,到


转载请注明:http://www.aierlanlan.com/rzfs/5150.html