AWS DLAMI - графические процессоры не отображаются для Keras

Question

Я создал DLAMI (Deep Learning AMI (Amazon Linux) версии 8.0 - ami-9109beee на g2.8xlarge) и установил ноутбук Jupyter для создания простого Keras LSTM. Когда я пытаюсь превратить свою модель Keras в модель GPU с помощью функции multi_gpu_model, я вижу следующую ошибку:

Игнорирование видимого устройства GPU (устройство: 0, имя: GRID K520, идентификатор шины PCI: 0000: 00: 03.0, вычислительные возможности: 3.0) с вычислительными возможностями Cuda 3.0. Минимальная требуемая способность Cuda составляет 3,5.

Я попытался переустановить tenorflow-gpu безрезультатно. Есть ли способ выровнять совместимость по этому AMI?

Answer 1

Эта проблема была решена путем удаления, а затем переустановки tenorflow-gpu через среду conda, предоставленную AMI.

Answer 2

Двоичные файлы TensorFlow, которые вы устанавливаете через pip или аналогичные, созданы только для поддержки вычислительных возможностей CUDA 3.5, но TensorFlow поддерживает вычислительные возможности 3.0.

К сожалению, единственный способ получить установку TensorFlow, которая поддерживает вычислительные возможности 3.0, - это сборка из источника .