Question

我最近开始在python上使用Tensorflow进行编程。我想在我的笔记本电脑中使用我的GTX 1050来提高计算能力，但我没有取得成功......

安装了所有必需的库和软件（CUDA 9.0，CuDNN for CUDA 9.0，导入：tensorflow和tensorflow-gpu ......）后，我尝试了Tensorflow网站的基本示例：

ListView

但这会返回以下答案（我在之后翻译）：

import tensorflow as tf
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)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))`

这个重要的消息意味着我的笔记本电脑上唯一可用的设备是我的CPU，但我有一台GTX 1050。我试着在开头添加这一行： 2018-05-03 19:15:59.540038: I T:\src\github\tensorflow\tensorflow\core\platform\cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 Device mapping: no known devices. 2018-05-03 19:15:59.543664: I T:\src\github\tensorflow\tensorflow\core\common_runtime\direct_session.cc:284] Device mapping: MatMul: (MatMul): /job:localhost/replica:0/task:0/device:CPU:0 2018-05-03 19:15:59.546930: I T:\src\github\tensorflow\tensorflow\core\common_runtime\placer.cc:886] MatMul: (MatMul)/job:localhost/replica:0/task:0/device:CPU:0 b: (Const): /job:localhost/replica:0/task:0/device:CPU:0 2018-05-03 19:15:59.547254: I T:\src\github\tensorflow\tensorflow\core\common_runtime\placer.cc:886] b: (Const)/job:localhost/replica:0/task:0/device:CPU:0 a: (Const): /job:localhost/replica:0/task:0/device:CPU:0 2018-05-03 19:15:59.547597: I T:\src\github\tensorflow\tensorflow\core\common_runtime\placer.cc:886] a: (Const)/job:localhost/replica:0/task:0/device:CPU:0 [[22. 28.] [49. 64.]] 我免除了你对命令行的所有侮辱，但它让我回答：

with tf.device("/device:gpu:0"):

有人知道我的问题的根源吗？或者有人曾经克服过这个问题，可以帮助我吗？

Answer 1

如果它可以帮助我删除tensorflow-gpu并通过pip命令再次安装它（pip install tensorflow -gpu == 1.8），我指定版本1.8并且它神奇地工作。

如何在GTX 1050移动版中使用Tensorflow？

1 个答案: