Я недавно начал программировать с Tensorflow на python. Я хотел улучшить вычислительную мощность, используя мой GTX 1050 в моем ноутбуке, но у меня ничего не вышло ...
После установки всех необходимых библиотек и программ (CUDA 9.0, CuDNN для CUDA 9.0, импортированные: tenorflow и tenorflow-gpu ...) я попробовал базовый пример с веб-сайта Tensorflow:
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))`
но это возвращает следующий ответ (я перевожу его сразу после):
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.]]
Это тяжелое сообщение означает, что единственным устройством, доступным на моем ноутбуке, является мой процессор, но у меня есть GTX 1050.
Я попытался добавить эту строку в начале:
with tf.device("/device:gpu:0"):
Я избавляю вас от всего оскорбления командной строки, но оно возвращает мне следующее:
Operation was explicitly assigned to /device:gpu:0 but available devices are
[ /job:localhost/replica:0/task:0/device:CPU:0 ].
Make sure the device specification refers to a valid device.
Кто-нибудь имеет представление о происхождении моей проблемы? Или кто-то когда-нибудь преодолевал эту проблему и мог мне помочь?