Как запустить TensorFlow на графическом процессоре AMD/ATI? ⇐ Python

[Гость]

После прочтения этого руководства https://www.tensorflow.org/guide/using_gpu я проверил сеанс графического процессора на этом простом коде

импортировать numpy как np импортировать matplotlib.pyplot как plt импортировать тензорный поток как 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], форма = [3,2], имя = 'b') c = tf.matmul(a, b) с tf.Session(config=tf.ConfigProto(log_device_placement=True)) как sess: х = sess.run(c) распечатать(х) Результат был

2018-08-07 18:44:59.019144: Я tensorflow/core/platform/cpu_feature_guard.cc:141] Ваш процессор поддерживает инструкции, для которых этот двоичный файл TensorFlow не был скомпилирован: AVX2 FMA Сопоставление устройств: известные устройства отсутствуют. 2018-08-07 18:44:59.019536: Я tensorflow/core/common_runtime/direct_session.cc:288] Сопоставление устройств: MatMul: (MatMul): /job:localhost/replica:0/task:0/device:CPU:0 2018-08-07 18:44:59.019902: Я tensorflow/core/common_runtime/placer.cc:886] MatMul: (MatMul)/job:localhost/replica:0/task:0/device:CPU:0 a: (Const): /job:localhost/replica:0/task:0/device:CPU:0 07.08.2018 18:44:59.019926: I tensorflow/core/common_runtime/placer.cc:886] а: (Const)/job:localhost/replica:0/task:0/device:CPU:0 b: (Const): /job:localhost/replica:0/task:0/device:CPU:0 07.08.2018 18:44:59.019934: I tensorflow/core/common_runtime/placer.cc:886] б: (Const)/job:localhost/replica:0/task:0/device:CPU:0 [[ 22. 28.] [ 49. 64.]] Как видите, графический процессор не выполняет никаких вычислений. и когда я изменил код, чтобы использовать конфигурацию графического процессора и долю процесса:

conf = tf.ConfigProto() conf.gpu_options.per_process_gpu_memory_fraction = 0,4 с tf.Session(config = conf) как sess: х = sess.run(c) распечатать(х) Результат был

2018-08-07 18:52:22.681221: Я tensorflow/core/platform/cpu_feature_guard.cc:141] Ваш процессор поддерживает инструкции, для которых этот двоичный файл TensorFlow не был скомпилирован: AVX2 ФМА [[ 22. 28.] [ 49. 64.]] Что я могу сделать, чтобы запустить сеанс на карте графического процессора?