Обновление до tensorflow2.0 alpha с некоторыми проблемами cuda

#tensorflow #cuda

Вопрос:

Я уже использую tensorflow1.12 и Cuda9.0 на своем компьютере с Windows. Сегодня я хочу обновить tensorflow до 2.0 alpha, поэтому я создаю новую среду с помощью Anaconda, а pip устанавливает tensorflow-gpu 2.0 alpha. Поскольку я знаю, что требуется cuda 10.0, пока я не хочу удалять существующую версию 9.0, поэтому я пытаюсь conda install cudatoolkit cudnn . Показано, что cudatoolkit: 10.0.130-0 и cudnn: 7.3.1-cuda10.0_0 загружаются автоматически.

Затем я пробую tensorflow2.0. import tensorflow ошибок не вызывает. Однако, когда я использую x=tf.constant([1]) , возникает это

 cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

Я думаю, проблема в том, что мой tensorflow2.0 не может найти загруженный cudatoolkit 10.0, хотя они находятся в той же среде anaconda. Итак, как я могу это решить

1. Сообщение об ошибке чрезвычайно простое и буквальное. Установленная вами версия драйвера графического процессора не поддерживает CUDA 10. Вам нужно обновить его

2. да, вы правы, я решаю эту проблему, обновляя версию GPU. В любом случае спасибо

Ответ №1:

Это можно легко решить, обновив мою версию драйвера GPU до самой новой.