tensorFlow中的nvidia cuda 7.5驱动程序未正确处理(ubuntu 14.04)

ecr0jaav  于 2023-03-03  发布在  其他
关注(0)|答案(2)|浏览(128)

在没有pb的安装后,我正在尝试有关GPU的教程:I类型:

with tf.device('/gpu:0'):
  a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
  b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
  c = tf.matmul(a, b)
  print(c)
  sess.run(c)

我得到了:
Tensor("MatMul_1:0",形状=Tensor形状([维度(2),维度(2)]),数据类型=浮点32,设备=/gpu:0)

追溯(最近调用最后调用):文件"/home/olivier/anaconda/lib/python2.7/站点包/tensorflow /python/client/www.example.com "中的第1行,运行结果= self._do_run中的第345行session.pyin_do_run e.代码)tensorflow . python.框架.错误.无效参数错误:无法将设备分配给节点"b_1":无法满足显式设备规范"/gpu:0 "节点:b_1 =常量类型= DT_FLOAT,值=Tensor,设备= "/gpu:0 "由操作u" b_1 "引起,定义位置:文件"/home/olivier/anaconda/lib/python2.7/site-packages/tensorflow/python/ops/constant_op. py "第147行常量属性={"值":Tensor值,"数据类型":d类型_值},名称=名称).输出[0]文件"/home/olivier/anaconda/lib/python2.7/站点包/tensorflow /python/框架/www.example.com ",第1710行,在创建操作中原始操作=自定义._默认原始操作,操作定义=操作定义)文件"/home/olivier/anaconda/lib/python2.7/站点包/tensorflow /python/框架/ops.py",第988行,在初始化自身中。回溯=提取堆栈() dtype_value}, name=name).outputs[0] File "/home/olivier/anaconda/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 1710, in create_op original_op=self._default_original_op, op_def=op_def) File "/home/olivier/anaconda/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 988, ininitself._traceback = _extract_stack()
在Torch7中,我的GPU工作正常

qvk1mo1f

qvk1mo1f1#

google发布的二进制文件需要在路径库中找到libcudart.so.7.0,你只需要像这样把它添加到LD_LIBRARY_PATH
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/home/olivier/digits-2.0/lib/cuda"
你放进.bashrc

cngwdvgl

cngwdvgl2#

在Optimus笔记本电脑(运行Manjaro Linux)上,可以通过optirun启动python控制台来运行TensorFlow和cuda加速:

$optirun python

我详细介绍了做这件事的方法。

相关问题