ValueError «Bạn đang cố gắng sử dụng back-end GPU cũ» khi nhập keras

Question

Tôi đang sử dụng Keras với phụ trợ Theano trên Ubuntu 16.04. Thiết lập của tôi đã được làm việc mà không vấn đề, tuy nhiên, tất cả của một đột ngột tôi nhận được lỗi sau khi tôi nhập Keras (import keras):

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.

Làm thế nào để giải quyết này? Xin vui lòng hỗ trợ, cảm ơn trước.

Answer 1

Bạn nên thay đổi (hoặc thêm) biến môi trường của bạn được gọi là THEANO_FLAGS. Nếu bạn đặt biến để nó chứa device=cuda thay vì device=gpu thì lỗi sẽ biến mất.

Thông tin thêm về biến này có thể được tìm thấy here.