centos7 tensorflow 1.13.1 . cuda 10 cudnn 7.3.1

tensorflow .马上要出 2.0 版本,不过现在 安装的最新的仍旧为1.13.1 【2019-03-9】
在 centos 7 我们安装了 一块 NVIDIA Tesla的 显卡 ,Tesla P100-PCIE-16GB

当我们要使用TensorFlow GPU 训练 时 , 单单有显卡还不可以,需要安装NVIDIA的显卡驱动 和cuda及cudnn,
尝试使用 原始的rpm 安装 cuda 发现 本来要安装 9 ,装完了竟然是10.1,TensorFlow 1.13.1 是不支持 10.1,不过现在日志看到 是支持 cuda 10.的
发现 cuda 的安装 还比较繁琐,cuda 安装 完占硬盘 2-3G
卸载还不是很方便,有的有一个卸载脚本 Perl,
/usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl
没有的话,就直接
rm -rf /usr/local/cuda*
rm /etc/profile.d/cuda.sh
也可以

那如何安装cuda 最简单就是 使用 conda ,
anaconda 简直不一般,conda不仅可以安装python虚拟环境 ,还可以安装R ,竟然连 cuda 和cudnn 也可以安装 ,还要安装 TensorFlow-gpu python包
就两步
首先先创建 激活一个python的虚拟环境 进入到环境中

conda install cudatoolkit
conda install cudnn
pip  install  tensorflow-gpu

不过要注意了 conda安装的cuda 并不是全局的cuda ,他只在你的当前激活的python环境下有效,conda 安装的jupyter 和R 也是这样的
参考 https://github.com/tensorflow/tensorflow/issues/26182
【ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory】
在安装时报了一个小错

ChunkedEncodingError(ProtocolError('Connection broken: OSError("(104, 'ECONNRESET')")', OSError("(104, 'ECONNRESET')")))
其实好像是网络问题,重复尝试再安装就可以了

如何测试是否 TensorFlow可以使用gpu 呢

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

查看日志信息若包含gpu信息,就是使用了gpu。 
其他方法:跑计算量大的代码,通过 nvidia-smi 命令查看gpu的内存使用量。

另外可以这样
import tensorflow as tf
 
with tf.device('/cpu:0'):
    a = tf.constant([1.0,2.0,3.0],shape=[3],name='a')
    b = tf.constant([1.0,2.0,3.0],shape=[3],name='b')
with tf.device('/gpu:1'):
    c = a+b
   
#注意:allow_soft_placement=True表明:计算设备可自行选择,如果没有这个参数,会报错。
#因为不是所有的操作都可以被放在GPU上,如果强行将无法放在GPU上的操作指定到GPU上,将会报错。
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True))
#sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
sess.run(tf.global_variables_initializer())
print(sess.run(c))

结果:

[ 2.  4.  6.]

如果TensorFlow不支持 当前 版本的cuda 会报错 【首先你要安装了TensorFlow-gpu包】

import tensorflow as tf
ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory

如何查看 GPU的使用情况

nvidia-smi

另外一点就是 cuda 再卸载后重新安装新版本 会报错
【# NVIDIA NVML Driver/library version mismatch

https://stackoverflow.com/questions/43022843/nvidia-nvml-driver-library-version-mismatch

As @etal said, rebooting can solve this problem, but I think a procedure without rebooting will help.

For Chinese, check my blog -> [中文版](https://comzyh.com/blog/archives/967/)

The error message

> NVML: Driver/library version mismatch

tell us the Nvidia driver kernel module (kmod) have a wrong version, so we should unload this driver, and then load the correct version of kmod

## How to do that ?

First, we should know which drivers are loaded.

> lsmod | grep nvidia

you may get

nvidia_uvm 634880 8
nvidia_drm 53248 0
nvidia_modeset 790528 1 nvidia_drm
nvidia 12312576 86 nvidia_modeset,nvidia_uvm


our final goal is to unload `nvidia` mod, so we should unload the module depend on `nvidia`

> sudo rmmod nvidia_drm
> sudo rmmod nvidia_modeset
> sudo rmmod nvidia_uvm

then, unload `nvidia`

> sudo rmmod nvidia

## Troubleshooting

if you get an error like `rmmod: ERROR: Module nvidia is in use`, which indicates that the kernel module is in use, you should kill the process that using the kmod:

> sudo lsof /dev/nvidia*

and then kill those process, then continue to unload the kmods

## Test

confirm you successfully unload those kmods

> lsmod | grep nvidia

you should get nothing, then confirm you can load the correct driver

> nvidia-smi

you should get the correct output

https://tensorflow.google.cn/install/source

https://comzyh.com/blog/archives/967/

如果你打算安装 cuda 9
https://blog.veir.me/2018/03/17/centos7-install-cuda9/

或者 cuda的docker环境
https://www.cnblogs.com/yxfangcs/p/8438462.html

cuda 的官方下载
https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=CentOS&target_version=7&target_type=rpmnetwork

TensorFlow 到底和那些 版本的cuda 和cudnn 适配 ,主要看 libcudnn.so libcublas.so 的当前安装TensorFlow所支持的版本
libcublas.so.10.0 libcudnn.so.7是TensorFlow 1.13.1所要求的 ,所以就必须装的版本不高于它

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,839评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,543评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,116评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,371评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,384评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,111评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,416评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,053评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,558评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,007评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,117评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,756评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,324评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,315评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,539评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,578评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,877评论 2 345

推荐阅读更多精彩内容