![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
TensorFlow
文章平均质量分 71
volcano_Lin
热爱CV和DL
展开
-
TensorFlow学习笔记[二]——在训练模型时如何指定具体哪一(多)块GPU显卡进行训练任务
在用tensorflow深度学习模型训练时,假设我们在训练之前没有指定具体用哪一块GPU进行训练,则默认的是选用第0块GPU来训练我们的模型。如果你的电脑有多块GPU的话,其它几块GPU的也会显示被占用。 有些时候,我们希望可以通过自己指定一块或者几块GPU来训练我们的模型,而不是用这种系统默认的方法。接下来将介绍三种指定GPU训练的方法。 我们...原创 2018-07-18 09:32:06 · 16509 阅读 · 8 评论 -
libcudart.so.8.0: cannot open shared object file: No such file or directory 的解决办法
在用GPU版本的Tensorfolow进行模型训练时,常常会出现 ImportError: libcudart.so.8.0: cannot open shared object file: No such file or directory 这样的错误。 首先检查/usr/local/cuda-8.0/lib64目录,看看libcudart.so.8.0是否存在,...原创 2018-07-11 22:21:30 · 62175 阅读 · 28 评论 -
TensorFlow学习笔记[一]—— 填充张量函数 tf.pad 解析
函数的作用:对张量按照你的需求进行填充定义在:tensorflow/python/ops/array_ops.py函数原型:pad( tensor, paddings, mode='CONSTANT', name=None)参数说明:tensor: A 'Tensor'. #tensor是要填充的张量;paddings: A 'Tensor' of ...原创 2018-07-14 09:38:31 · 4586 阅读 · 0 评论 -
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
问题:今天在安装imgaug数据增强库的时候,提示TensorFlow需要1.7.0版本 升级tensorflow版本之后,import tensorflow的时候出现如下错误:ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory 原因...原创 2018-09-05 11:13:56 · 17813 阅读 · 3 评论