Как я могу решить ошибку theano GPU pygpu в Google Colab?

Question

Я использую theano GPU в Google Colab и получаю эту ошибку:

ERROR (theano.gpuarray): pygpu was configured but could not be imported or is too old (version 0.7 or higher required)

Я установил:

import os
os.environ["THEANO_FLAGS"]="device=cuda, floatX=float32"

И изменил тип времени выполнения на аппаратный ускоритель 'GPU '

Не могли бы вы помочь мне решить эту проблему?

Спасибо.