2017-06-26 10 views
7

Używam Keras z serwerem Theano w systemie Ubuntu 16.04. Moja konfiguracja pracuje bez problemów, jednak nagle pojawia się następujący komunikat o błędzie podczas importowania Keras (import keras):ValueError «Starasz się używać starego back-end'u GPU» podczas importowania kabli

ValueError: You are trying to use the old GPU back-end. It was removed from Theano. Use device=cuda* now. See https://github.com/Theano/Theano/wiki/Converting-to-the-new-gpu-back-end%28gpuarray%29 for more information.

Jak mogę rozwiązać ten problem? Proszę, dziękuję z góry.

+2

Jest to prawdopodobnie zmienna środowiskowa "THEANO_FLAGS". Jeśli ustawisz zmienną tak, że zawiera ona 'device = cuda' zamiast' device = gpu', to zostanie ona naprawiona. –

+0

Witam @ WilmarvanOmmeren, właśnie wypróbowałem to teraz i zadziałało. Jeśli umieścisz swój komentarz jako odpowiedź, oznaczy go jako poprawną odpowiedź. Dzięki . –

Odpowiedz

11

Należy zmienić (lub dodać) zmienną środowiskową o nazwie THEANO_FLAGS. Jeśli ustawisz zmienną tak, aby zawierała device=cuda zamiast device=gpu, błąd zniknie.

Więcej informacji na temat tej zmiennej można znaleźć here.

+1

eksportuj THEANO_FLAGS = tryb = FAST_RUN, urządzenie = cuda, floatX = float32 – cgl

Powiązane problemy