Запуск примера Keras MNIST в Ubuntu 18.04 - ошибка графического процессора - PullRequest
Новая
       31

Запуск примера Keras MNIST в Ubuntu 18.04 - ошибка графического процессора

0 голосов
/ 17 марта 2019

Я запускал следующий код Keras / TensorFlow после установки Anaconda 2018.12 в Ubuntu 18.04.У меня есть драйвер Nvidia 390.116 и графический процессор GTX 1070.Однако, когда я пытаюсь запустить этот простой пример MNIST 

import tensorflow as tf
mnist = tf.keras.datasets.mnist

(x_train, y_train),(x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0

model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(512, activation=tf.nn.relu),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10, activation=tf.nn.softmax)
])
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

model.fit(x_train, y_train, epochs=5)
model.evaluate(x_test, y_test)

, я получаю трассировку, которая указывает на недостаточно новый драйвер NVidia.Вся трассировка длинная, но я думаю, что ключевая строка -

File "/home/christopher/anaconda3/envs/tensorflow_gpuenv/lib/python3.6/site-packages/tensorflow/python/client/session.py ", строка 676, в init self._session = tf_session.TF_NewSessionRef (self._graph._c_graph, opts) tenorflow.python.framework.errors_impl.InternalError: cudaGetDevice () не удалось.Состояние: версии драйвера CUDA недостаточно для версии времени выполнения CUDA

Это, кажется, не имеет смысла, так как это самый последний из доступных драйверов для Ubuntu 18.04.Это не является дубликатом ранее недостаточной проблемы с драйвером, потому что я использую драйвер Nvidia 390.116.

1 Ответ

1 голос
/ 18 марта 2019

Я решил эту проблему, посмотрев статью о системах pugets, упомянутую выше. Однако очистка старого драйвера и установка предложенного драйвера из этой статьи не помогли мне. Однако после очистки исходного драйвера Nvidia (390.116), переключения на Nouveau, перезагрузки и установки драйвера Nvidia 415.27 с помощью страницы свойств «Дополнительные драйверы» в приложении Software and Sources я смог успешно запустить пример кода.

...